Updated the test suite for the current API.

1 files changed, 4 insertions, 3 deletions

diff --git a/tests/analysis/contents/memory.py b/tests/analysis/contents/memory.py
index e8ba6b5..55ce035 100644
--- a/tests/analysis/contents/memory.py
+++ b/tests/analysis/contents/memory.py
@@ -7,6 +7,7 @@
 
 
 from chrysacase import ChrysalideTestCase
+from pychrysalide import arch
 from pychrysalide.analysis.contents import MemoryContent
 from pychrysalide.arch import vmpa, mrange
 
@@ -38,12 +39,12 @@ class TestMemoryContent(ChrysalideTestCase):
 
         start = vmpa(14, vmpa.VMPA_NO_VIRTUAL)
 
-        val = cnt.read_u16(start, vmpa.SRE_LITTLE)
+        val = cnt.read_u16(start, arch.SRE_LITTLE)
         self.assertEqual(val, 0x1817)
 
         start = vmpa(10, vmpa.VMPA_NO_VIRTUAL)
 
-        val = cnt.read_u32(start, vmpa.SRE_LITTLE)
+        val = cnt.read_u32(start, arch.SRE_LITTLE)
         self.assertEqual(val, 0x16150013)
 
 
@@ -62,4 +63,4 @@ class TestMemoryContent(ChrysalideTestCase):
         with self.assertRaisesRegex(Exception, 'Invalid read access.'):
 
             start = vmpa(0, vmpa.VMPA_NO_VIRTUAL)
-            val = cnt.read_u16(start, vmpa.SRE_LITTLE)
+            val = cnt.read_u16(start, arch.SRE_LITTLE)