diff options
Diffstat (limited to 'tests/analysis/contents/memory.py')
| -rw-r--r-- | tests/analysis/contents/memory.py | 7 |
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) |