pocketpy/tests/01_int.py

# test int literals
assert 0xffff == 65535
assert 0xAAFFFF == 11206655
assert 0x7fffffff == 2147483647
assert -0xffff == -65535
assert -0xAAFFFF == -11206655
assert -0x7fffffff == -2147483647
# test 64-bit
assert 2**60-1 + 546 - 0xfffffffffffff == 1148417904979477026

# test oct literals
assert 0o1234 == 668
assert 0o17777777777 == 2147483647
assert -0o1234 == -668
assert -0o17777777777 == -2147483647

# test binary literals
assert 0b10010 == 18
assert -0b10010 == -18
assert 0b11111111111111111111111111111111 == 4294967295
assert -0b11111 == -31

# test == != >= <= < >
assert -1 == -1
assert -1 != 1
assert -1 >= -1
assert -1 <= -1
assert -1 < 1
assert -1 > -2

# test + - * % ** //
assert -1 + 1 == 0
assert -1 - 1 == -2
assert 4 * -1 == -4
assert 5 % 2 == 1
assert 2 ** 3 == 8
assert 4 // 2 == 2
assert 5 // 2 == 2

# test += -= *= //=
x = 3
x += 1
assert x == 4
x -= 1
assert x == 3
x *= 2
assert x == 6
x //= 2
assert x == 3

# test bit_length
assert (1).bit_length() == 1
assert (2).bit_length() == 2
assert (3).bit_length() == 2

assert (-1).bit_length() == 1
assert (-2).bit_length() == 2
assert (-3).bit_length() == 2

assert (123123123123123).bit_length() == 47
assert (-3123123123).bit_length() == 32

# test int()
assert int() == 0
assert int(True) == 1
assert int(False) == 0

assert int(1) == 1
assert int(1.0) == 1
assert int(1.1) == 1
assert int(1.9) == 1
assert int(-1.9) == -1
assert int(1.5) == 1
assert int(-1.5) == -1
assert int("123") == 123

assert int("0x123", 16) == 291
assert int("0o123", 8) == 83
assert int("-0x123", 16) == -291
assert int("-0o123", 8) == -83
assert int("-123") == -123
assert int("+123") == 123

# test >> << & | ^
assert 12 >> 1 == 6
assert 12 << 1 == 24
assert 12 & 1 == 0
assert 12 | 1 == 13
assert 12 ^ 1 == 13

# test high precision int pow
assert 7**21 == 558545864083284007
assert 2**60 == 1152921504606846976
assert -2**60 == -1152921504606846976
assert 4**13 == 67108864
assert (-4)**13 == -67108864

assert ~3 == -4
assert ~-3 == 2
assert ~0 == -1

# test __str__, __repr__
assert str(1) == '1'
assert repr(1) == '1'

assert 1 < 2 < 3
assert 4 > 3 >= 3
assert not 1 < 2 > 3

# try:
#     1 // 0
#     exit(1)
# except ZeroDivisionError:
#     pass

# try:
#     1 % 0
#     exit(1)
# except ZeroDivisionError:
#     pass

# try:
#     2**60 // 0
#     exit(1)
# except ZeroDivisionError:
#     pass

# try:
#     2**60 % 0
#     exit(1)
# except ZeroDivisionError:
#     pass

# try:
#     divmod(1, 0)
#     exit(1)
# except ZeroDivisionError:
#     pass

# try:
#     divmod(2**60, 0)
#     exit(1)
# except ZeroDivisionError:
#     pass

# try:
#     x = eval("231231312312312312312312312312312312314354657553423345632")
#     print(f"eval should fail, but got {x!r}")
#     exit(1)
# except SyntaxError:
#     pass