Add MemoryScanner and bcd tests

tests/test_bcd_conversion.py

@@ -0,0 +1,100 @@
+from pyboy import PyBoy, bcd_to_dec, dec_to_bcd
+from pyboy.api.memoryscanner import DynamicComparisonType
+
+
+def test_bcd_to_dec_single_byte():
+    assert bcd_to_dec(0b00110000) == 30
+
+
+def test_dec_to_bcd_single_byte():
+    assert dec_to_bcd(30) == 0b00110000
+
+
+def test_bcd_to_dec_multi_byte():
+    assert bcd_to_dec(0b0011000000110000, byte_width=2) == 3030
+
+
+def test_dec_to_bcd_multi_byte():
+    assert dec_to_bcd(3030, byte_width=2) == 0b0011000000110000
+
+
+def test_bcd_to_dec_complex():
+    assert bcd_to_dec(0b011101100101010010001001, byte_width=3) == 765489
+
+
+def test_dec_to_bcd_complex():
+    assert dec_to_bcd(765489, byte_width=3) == 0b011101100101010010001001
+
+
+def test_bcd_to_dec_complex2():
+    assert bcd_to_dec(0b10000000000000000000000000000000001000000, byte_width=6, byteorder="little") == 10000000040
+
+
+def test_memoryscanner_basic(kirby_rom):
+    pyboy = PyBoy(kirby_rom, window_type="null", game_wrapper=True)
+    pyboy.set_emulation_speed(0)
+    kirby = pyboy.game_wrapper
+    kirby.start_game()
+    addresses = pyboy.memory_scanner.scan_memory()
+    assert len(addresses) == 0x10000
+    addresses = pyboy.memory_scanner.scan_memory(start_addr=0xC000, end_addr=0xDFFF)
+    assert len(addresses) == 0xDFFF - 0xC000 + 1
+    addresses = pyboy.memory_scanner.rescan_memory(None, DynamicComparisonType.INCREASED)
+    assert len(addresses) == 0
+
+
+score_100 = 0xD072
+
+
+def test_memoryscanner_absolute(kirby_rom):
+    pyboy = PyBoy(kirby_rom, window_type="null", game_wrapper=True)
+    pyboy.set_emulation_speed(0)
+    kirby = pyboy.game_wrapper
+    kirby.start_game()
+
+    # Find score address
+    pyboy.tick(1, True)
+    pyboy.button_press("right")
+
+    pyboy.tick(175, True)
+    assert pyboy.memory[score_100] == 4
+    addresses = pyboy.memory_scanner.scan_memory(4, start_addr=0xC000, end_addr=0xDFFF)
+    assert score_100 in addresses
+    values = [pyboy.memory[x] for x in addresses]
+    assert all([x == 4 for x in values])
+
+    pyboy.tick(175, True)
+    assert pyboy.memory[score_100] == 8
+    addresses = pyboy.memory_scanner.rescan_memory(8, DynamicComparisonType.MATCH)
+    values = [pyboy.memory[x] for x in addresses]
+    assert all([x == 8 for x in values])
+
+    # Actual score address for third digit
+    assert score_100 in addresses
+    assert pyboy.memory[score_100] == 8
+
+
+def test_memoryscanner_relative(kirby_rom):
+    pyboy = PyBoy(kirby_rom, window_type="null", game_wrapper=True)
+    pyboy.set_emulation_speed(0)
+    kirby = pyboy.game_wrapper
+    kirby.start_game()
+
+    # Find score address
+    pyboy.tick(1, True)
+    score = 0
+    addresses = pyboy.memory_scanner.scan_memory(score, start_addr=0xC000, end_addr=0xDFFF)
+    assert len(addresses) > 1000, "We expected more values to be 0"
+    pyboy.button_press("right")
+
+    pyboy.tick(175, True)
+    addresses1 = pyboy.memory_scanner.rescan_memory(None, DynamicComparisonType.INCREASED)
+    values1 = [pyboy.memory[x] for x in addresses1]
+
+    pyboy.tick(175, True)
+    addresses2 = pyboy.memory_scanner.rescan_memory(None, DynamicComparisonType.INCREASED)
+    values2 = [pyboy.memory[x] for x in addresses2]
+
+    # Actual score address for third digit
+    assert score_100 in addresses2
+    assert pyboy.memory[score_100] == 8