Code: Select all
for A in range(0, 256):
for B in range(0, 256):
carry_in = 1
# Add with carry
C, carry_out, overflow = add(A, B, carry_in)
if not A[7] and not B[7] and carry_in:
assert C[7] == 1
assert overflow == 1
assert carry_out == 0
else if A[7] and B[7] and not carry_in:
assert C[7] == 0
assert overflow == 1
assert carry_out == 1
else:
assert overflow == 0
assert C == A + B
if C > 255:
assert carry_out == 1
Thanks.