structo 0.0.5__tar.gz → 0.0.6__tar.gz

{structo-0.0.5 → structo-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: structo
-Version: 0.0.5
+Version: 0.0.6
 Summary: Structify
 Author-email: Ben Brady <benbradybusiness@gmail.com>
 License-Expression: MIT

{structo-0.0.5 → structo-0.0.6}/structo/__init__.py

@@ -2,7 +2,7 @@
 Structify
 """
 
-__version__ = "0.0.5"
+__version__ = "0.0.6"
 
 from .serializer import Serializer, Serialiable
 from .serialise import get_serializer

{structo-0.0.5 → structo-0.0.6}/structo/serialise.py

@@ -12,7 +12,7 @@ def get_serializer(format: type) -> Serializer:
 
         return serializer
 
-    if isinstance(format, Serialiable):
+    if issubclass(format, Serialiable):
         return format.serializer()
 
     # Nicely formatted errors:

structo-0.0.6/tests/test_object.py

@@ -0,0 +1,54 @@
+from utils import test
+import typing as t
+
+import structo as st
+
+
+@test("SerialiableObject: object")
+def _():
+    class Foo(st.SerializableObject):
+        a: t.Annotated[int, st.uint8]
+
+    obj = Foo(a=1)
+    assert obj == Foo.from_bytes(obj.to_bytes())
+
+
+@test("SerialiableObject: multiple attributes")
+def _():
+    class Foo(st.SerializableObject):
+        a: t.Annotated[int, st.uint8]
+        b: t.Annotated[int, st.uint16_BE]
+
+    obj = Foo(a=1, b=256)
+    assert obj == Foo.from_bytes(obj.to_bytes())
+
+
+@test("SerialiableObject: sizeof")
+def _():
+
+    class Foo(st.SerializableObject):
+        a: t.Annotated[int, st.uint8]
+        b: t.Annotated[str, st.uint8]
+
+    assert Foo.sizeof() == 2
+
+
+@test("SerialiableObject: sizeof unknowable")
+def _():
+    class Foo(st.SerializableObject):
+        a: t.Annotated[int, st.uint8]
+        b: t.Annotated[str, st.String(st.uint8)]
+
+    assert Foo.sizeof() is None
+
+
+@test("SerialiableObject: nested object")
+def _():
+    class Foo(st.SerializableObject):
+        a: t.Annotated[int, st.uint8]
+
+    class Bar(st.SerializableObject):
+        foo: Foo
+
+    obj = Bar(foo=Foo(1))
+    assert obj == Bar.from_bytes(obj.to_bytes())