def test_little_endian():
rom_bytes = bytearray(b"\x00\x01\x02\x03\x04\x05\x06\x00")
rom = Rom(rom_bytes)
assert rom.little_endian(0) == (0x01 << 8) + 0x00
assert rom.little_endian(6) == (0x00 << 8) + 0x06
def test_find():
rom_bytes = bytearray(b"\x00\x01\x02\x03\x04\x05\x06\x00")
rom = Rom(rom_bytes)
assert rom.find(b"\x00") == 0
assert rom.find(b"\x00", 1) == 7
assert rom.find(b"\x07") == -1
def test_int():
rom_bytes = bytearray(b"\x00\x01\x02\x03\x04\x1f")
numbers = [0, 1, 2, 3, 4, 0x1F]
rom = Rom(rom_bytes)
for offset, number in enumerate(numbers):
assert rom.int(offset) == number
def rom():
if not test_rom_path.exists():
raise ValueError(
f"To run the test suite, place a US SMB3 Rom named '{test_rom_path}' in the root of the repository."
)
with open(test_rom_path, "rb") as rom_file:
yield Rom(bytearray(rom_file.read()))
def rom():
with open("SMB3.nes", "rb") as rom_file:
yield Rom(bytearray(rom_file.read()))