mirror of
https://github.com/TheAlgorithms/Python.git
synced 2026-09-28 13:33:11 +08:00
Poorti GuptaandChristian Clauss
0946da0119
* Bit manipulation: add type hints, validation, doctests; simplify bitwise operations * Fix error message for non-integer inputs --------- Co-authored-by: Christian Clauss <cclauss@me.com>
91 lines
2.2 KiB
Python
91 lines
2.2 KiB
Python
"""Bitwise AND helper.
|
|
|
|
Return a zero-padded binary string representing ``a & b`` where the width is the
|
|
maximum bit length of the inputs. Only non-negative integers are accepted.
|
|
|
|
>>> binary_and(25, 32)
|
|
'0b000000'
|
|
>>> binary_and(37, 50)
|
|
'0b100000'
|
|
>>> binary_and(21, 30)
|
|
'0b10100'
|
|
>>> binary_and(58, 73)
|
|
'0b0001000'
|
|
>>> binary_and(0, 255)
|
|
'0b00000000'
|
|
>>> binary_and(256, 256)
|
|
'0b100000000'
|
|
|
|
Invalid inputs raise clear exceptions:
|
|
|
|
>>> binary_and(0, -1)
|
|
Traceback (most recent call last):
|
|
...
|
|
ValueError: inputs must be non-negative integers
|
|
>>> binary_and(0, 1.1)
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: inputs must be integers
|
|
>>> binary_and("0", "1")
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: inputs must be integers
|
|
"""
|
|
|
|
# https://www.tutorialspoint.com/python3/bitwise_operators_example.htm
|
|
|
|
|
|
def binary_and(a: int, b: int) -> str:
|
|
"""
|
|
Take in 2 integers, convert them to binary,
|
|
return a binary number that is the
|
|
result of a binary and operation on the integers provided.
|
|
|
|
>>> binary_and(25, 32)
|
|
'0b000000'
|
|
>>> binary_and(37, 50)
|
|
'0b100000'
|
|
>>> binary_and(21, 30)
|
|
'0b10100'
|
|
>>> binary_and(58, 73)
|
|
'0b0001000'
|
|
>>> binary_and(0, 255)
|
|
'0b00000000'
|
|
>>> binary_and(256, 256)
|
|
'0b100000000'
|
|
>>> binary_and(0, -1)
|
|
Traceback (most recent call last):
|
|
...
|
|
ValueError: inputs must be non-negative integers
|
|
>>> binary_and(0, 1.1)
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: inputs must be integers
|
|
>>> binary_and("0", "1")
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: inputs must be integers
|
|
>>> binary_and(10, "1")
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: inputs must be integers
|
|
"""
|
|
if not isinstance(a, int) or not isinstance(b, int):
|
|
raise TypeError("inputs must be integers")
|
|
if a < 0 or b < 0:
|
|
raise ValueError("inputs must be non-negative integers")
|
|
|
|
a_binary = format(a, "b")
|
|
b_binary = format(b, "b")
|
|
|
|
max_len = max(len(a_binary), len(b_binary))
|
|
|
|
max_len = max(a.bit_length(), b.bit_length())
|
|
return f"0b{(a & b):0{max_len}b}"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import doctest
|
|
|
|
doctest.testmod()
|