Fix compare with None

Author: andreax79

changelog.txt

@@ -119,3 +119,11 @@
 - Python 3.10 support
 - pytest
 - black code style
+
+### 2.3
+
+2022-09-01
+
+### Fix
+
+- Fix compare with None

cstruct/__init__.py

@@ -26,7 +26,7 @@
 
 __author__ = 'Andrea Bonomi <andrea.bonomi@gmail.com>'
 __license__ = 'MIT'
-__version__ = '2.2'
+__version__ = '2.3'
 __date__ = '15 August 2013'
 
 import struct

cstruct/abstract.py

@@ -141,7 +141,7 @@ def size(self) -> int:
         return self.__size__
 
     def __eq__(self, other: Any) -> bool:
-        return isinstance(other, self.__class__) and self.__dict__ == other.__dict__
+        return other is not None and isinstance(other, self.__class__) and self.__dict__ == other.__dict__
 
     def __ne__(self, other: Any) -> bool:
         return not self.__eq__(other)

tests/test_cstruct.py

@@ -246,3 +246,9 @@ def test_nested():
     assert flat.endAddrCylSec == nested.end.addrCylSec
     assert flat.startLBA == nested.startLBA
     assert flat.endLBA == nested.endLBA
+
+
+def test_null_compare():
+    c = Dummy()
+    assert c is not None
+    assert c != None

tests/test_memcstruct.py

@@ -243,3 +243,9 @@ def test_nested():
     assert flat.endAddrCylSec == nested.end.addrCylSec
     assert flat.startLBA == nested.startLBA
     assert flat.endLBA == nested.endLBA
+
+
+def test_null_compare():
+    c = Dummy()
+    assert c is not None
+    assert c != None