86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
import pytest
|
|
from src.basic_data_handling.int_nodes import (
|
|
IntAdd,
|
|
IntSubtract,
|
|
IntMultiply,
|
|
IntDivide,
|
|
IntModulus,
|
|
IntPower,
|
|
IntBitLength,
|
|
IntToBytes,
|
|
IntFromBytes,
|
|
IntBitCount,
|
|
)
|
|
|
|
|
|
def test_int_add():
|
|
node = IntAdd()
|
|
assert node.add(10, 5) == (15,)
|
|
assert node.add(-10, 5) == (-5,)
|
|
assert node.add(0, 0) == (0,)
|
|
|
|
|
|
def test_int_subtract():
|
|
node = IntSubtract()
|
|
assert node.subtract(10, 5) == (5,)
|
|
assert node.subtract(5, 10) == (-5,)
|
|
assert node.subtract(0, 0) == (0,)
|
|
|
|
|
|
def test_int_multiply():
|
|
node = IntMultiply()
|
|
assert node.multiply(10, 5) == (50,)
|
|
assert node.multiply(-1, 5) == (-5,)
|
|
assert node.multiply(0, 5) == (0,)
|
|
|
|
|
|
def test_int_divide():
|
|
node = IntDivide()
|
|
assert node.divide(10, 2) == (5,)
|
|
assert node.divide(9, 4) == (2,) # Integer division
|
|
with pytest.raises(ValueError, match="Cannot divide by zero."):
|
|
node.divide(10, 0)
|
|
|
|
|
|
def test_int_modulus():
|
|
node = IntModulus()
|
|
assert node.modulus(10, 3) == (1,)
|
|
assert node.modulus(9, 9) == (0,)
|
|
with pytest.raises(ValueError, match="Cannot perform modulus operation by zero."):
|
|
node.modulus(10, 0)
|
|
|
|
|
|
def test_int_power():
|
|
node = IntPower()
|
|
assert node.power(2, 3) == (8,)
|
|
assert node.power(10, 0) == (1,) # Any number to the power 0 is 1
|
|
assert node.power(-2, 3) == (-8,)
|
|
|
|
|
|
def test_int_bit_length():
|
|
node = IntBitLength()
|
|
assert node.bit_length(0) == (0,) # Zero requires 0 bits
|
|
assert node.bit_length(1) == (1,)
|
|
assert node.bit_length(255) == (8,) # 255 requires 8 bits
|
|
assert node.bit_length(-255) == (8,) # Negative numbers have the same bit length
|
|
|
|
|
|
def test_int_to_bytes():
|
|
node = IntToBytes()
|
|
assert node.to_bytes(255, 2, "big", "False") == (b"\x00\xff",)
|
|
assert node.to_bytes(255, 2, "little", "False") == (b"\xff\x00",)
|
|
assert node.to_bytes(-255, 2, "big", "True") == (b"\xff\x01",)
|
|
|
|
|
|
def test_int_from_bytes():
|
|
node = IntFromBytes()
|
|
assert node.from_bytes(b"\x00\xff", "big", "False") == (255,)
|
|
assert node.from_bytes(b"\xff\x00", "little", "False") == (255,)
|
|
assert node.from_bytes(b"\xff\x01", "big", "True") == (-255,)
|
|
|
|
|
|
def test_int_bit_count():
|
|
node = IntBitCount()
|
|
assert node.bit_count(0) == (0,) # Zero has no 1 bits
|
|
assert node.bit_count(255) == (8,) # 255 is 11111111 (8 ones)
|
|
assert node.bit_count(-255) == (8,) # Negative integer has the same bit count |