Refactor tests to conform to pytest asserts

1 files changed, 20 insertions, 13 deletions

diff --git a/tests/test_sparse_memory.py b/tests/test_sparse_memory.py
index 79b6638..7dda8f0 100644
--- a/tests/test_sparse_memory.py
+++ b/tests/test_sparse_memory.py
@@ -8,24 +8,31 @@ def mem():
 
 def test_oob_read(mem):
     for addr in range(mem.page_size):
-        with pytest.raises(MemoryAccessError): mem.read(addr, 1)
-        with pytest.raises(MemoryAccessError): mem.read(addr, 30)
-        with pytest.raises(MemoryAccessError): mem.read(addr + 0x10, 30)
-        with pytest.raises(MemoryAccessError): mem.read(addr, mem.page_size)
-        with pytest.raises(MemoryAccessError): mem.read(addr, mem.page_size - 1)
-        with pytest.raises(MemoryAccessError): mem.read(addr, mem.page_size + 1)
+        with pytest.raises(MemoryAccessError):
+            mem.read(addr, 1)
+        with pytest.raises(MemoryAccessError): 
+            mem.read(addr, 30)
+        with pytest.raises(MemoryAccessError): 
+            mem.read(addr + 0x10, 30)
+        with pytest.raises(MemoryAccessError): 
+            mem.read(addr, mem.page_size)
+        with pytest.raises(MemoryAccessError): 
+            mem.read(addr, mem.page_size - 1)
+        with pytest.raises(MemoryAccessError): 
+            mem.read(addr, mem.page_size + 1)
 
 def test_basic_read_write(mem):
     data = b'a' * mem.page_size * 2
     mem.write(0x300, data)
-    assert mem.read(0x300 == len(data), data)
-    assert mem.read(0x300 == 1, b'a')
-    assert mem.read(0x400 == 1, b'a')
-    assert mem.read(0x299 + mem.page_size * 2 == 1, b'a')
-    assert mem.read(0x321 == 12, b'aaaaaaaaaaaa')
+    assert mem.read(0x300, len(data)) == data
+    assert mem.read(0x300, 1) == b'a'
+    assert mem.read(0x400, 1) == b'a'
+    assert mem.read(0x299 + mem.page_size * 2, 1) == b'a'
+    assert mem.read(0x321, 12) == b'aaaaaaaaaaaa'
 
     mem.write(0x321, b'Hello World!')
-    assert mem.read(0x321 == 12, b'Hello World!')
+    assert mem.read(0x321, 12) == b'Hello World!'
 
-    with pytest.raises(MemoryAccessError): mem.read(0x300, mem.page_size * 3)
+    with pytest.raises(MemoryAccessError): 
+        mem.read(0x300, mem.page_size * 3)