Fix bug, use the correct type as key in the memory dict

Xilinx · Dec 3, 2021 · 207d9b0 · 207d9b0
1 parent 1d6dad1
commit 207d9b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/mock_device.py b/tests/mock_device.py
@@ -25,7 +25,7 @@ def write_register(self, address, data):
         self.memory[str(address)] = data
 
     def read_register(self, address):
-        return self.memory.get(address)
+        return self.memory.get(str(address))
 
 
 class MockRegisterIP(MockIPBase):