about summary refs log tree commit diff stats
path: root/tests/test_sparse_memory.py
diff options
context:
space:
mode:
authorTheofilos Augoustis <theofilos.augoustis@tum.de>2025-08-27 15:44:57 +0000
committerTheofilos Augoustis <theofilos.augoustis@tum.de>2025-08-27 16:46:29 +0000
commit9729ae38f8e44a17cb0915952791b5b906b4b7fa (patch)
treee2e0dd02d4eae136e7475346ab4f09c18930876e /tests/test_sparse_memory.py
parent849e5a6ec6e0246b5dde1fb2583aa13ed288e9c1 (diff)
downloadfocaccia-9729ae38f8e44a17cb0915952791b5b906b4b7fa.tar.gz
focaccia-9729ae38f8e44a17cb0915952791b5b906b4b7fa.zip
Migrate to nix flakes, uv and pyproject toml for builds
Diffstat (limited to 'tests/test_sparse_memory.py')
-rw-r--r--tests/test_sparse_memory.py33
1 files changed, 33 insertions, 0 deletions
diff --git a/tests/test_sparse_memory.py b/tests/test_sparse_memory.py
new file mode 100644
index 0000000..4fd9cba
--- /dev/null
+++ b/tests/test_sparse_memory.py
@@ -0,0 +1,33 @@
+import unittest
+
+from focaccia.snapshot import SparseMemory, MemoryAccessError
+
+class TestSparseMemory(unittest.TestCase):
+    def test_oob_read(self):
+        mem = SparseMemory()
+        for addr in range(mem.page_size):
+            self.assertRaises(MemoryAccessError, mem.read, addr, 1)
+            self.assertRaises(MemoryAccessError, mem.read, addr, 30)
+            self.assertRaises(MemoryAccessError, mem.read, addr + 0x10, 30)
+            self.assertRaises(MemoryAccessError, mem.read, addr, mem.page_size)
+            self.assertRaises(MemoryAccessError, mem.read, addr, mem.page_size - 1)
+            self.assertRaises(MemoryAccessError, mem.read, addr, mem.page_size + 1)
+
+    def test_basic_read_write(self):
+        mem = SparseMemory()
+
+        data = b'a' * mem.page_size * 2
+        mem.write(0x300, data)
+        self.assertEqual(mem.read(0x300, len(data)), data)
+        self.assertEqual(mem.read(0x300, 1), b'a')
+        self.assertEqual(mem.read(0x400, 1), b'a')
+        self.assertEqual(mem.read(0x299 + mem.page_size * 2, 1), b'a')
+        self.assertEqual(mem.read(0x321, 12), b'aaaaaaaaaaaa')
+
+        mem.write(0x321, b'Hello World!')
+        self.assertEqual(mem.read(0x321, 12), b'Hello World!')
+
+        self.assertRaises(MemoryAccessError, mem.read, 0x300, mem.page_size * 3)
+
+if __name__ == '__main__':
+    unittest.main()