Fix unit tests

fox-it · Aug 29, 2023 · 62e7d6f · 62e7d6f
1 parent 83db5b7
commit 62e7d6f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_align.py b/tests/test_align.py
@@ -289,7 +289,7 @@ def test_align_pointer():
     c = cstruct.cstruct(pointer="uint64")
     c.load(d, align=True)
 
-    assert c.pointer == "uint64"
+    assert c.pointer is c.uint64
 
     fields = c.test.fields
     assert c.test.align

diff --git a/tests/test_pointer.py b/tests/test_pointer.py
@@ -18,7 +18,7 @@ def test_pointer_basic(compiled):
     cs.load(cdef, compiled=compiled)
 
     assert verify_compiled(cs.ptrtest, compiled)
-    assert cs.pointer == "uint16"
+    assert cs.pointer is cs.uint16
 
     buf = b"\x04\x00\x08\x00\x01\x02\x03\x04\x05\x06\x07\x08"
     obj = cs.ptrtest(buf)
@@ -64,7 +64,7 @@ def test_pointer_struct(compiled):
 
     assert verify_compiled(cs.test, compiled)
     assert verify_compiled(cs.ptrtest, compiled)
-    assert cs.pointer == "uint16"
+    assert cs.pointer is cs.uint16
 
     buf = b"\x02\x00testt\x00e\x00s\x00t\x00\x01\x02\x03\x04\x05\x06\x07lalala\x00t\x00e\x00s\x00t\x00\x00\x00"
     obj = cs.ptrtest(buf)
@@ -97,7 +97,7 @@ def test_array_of_pointers(compiled):
     cs.load(cdef, compiled=compiled)
 
     assert verify_compiled(cs.mainargs, compiled)
-    assert cs.pointer == "uint16"
+    assert cs.pointer is cs.uint16
 
     buf = b"\x02\x09\x00\x16\x00\x00\x00\x00\x00argument one\x00argument two\x00"
     obj = cs.mainargs(buf)