UPDATE. only allow str/bytes value for Address.from_primitive()

Author: daehan-koreapool

pycardano/address.py

@@ -343,7 +343,18 @@ def to_primitive(self) -> bytes:
         return bytes(self)
 
     @classmethod
-    def from_primitive(cls: Type[Address], value: Union[bytes, str]) -> Address:
+    def from_primitive(cls: Type[Address], value: Primitive) -> Address:
+        if not isinstance(
+            value,
+            (
+                bytes,
+                str,
+            ),
+        ):
+            raise DeserializeException(
+                f"A bytes or a string value is required for deserialization: {value}"
+            )
+
         if isinstance(value, str):
             value = bytes(decode(value))
         header = value[0]

test/pycardano/test_address.py

@@ -24,3 +24,21 @@ class PointerAddressTest(TestCase):
     def test_from_primitive_invalid_value(self):
         with self.assertRaises(DeserializeException):
             PointerAddress.from_primitive(1)
+
+        with self.assertRaises(DeserializeException):
+            PointerAddress.from_primitive([])
+
+        with self.assertRaises(DeserializeException):
+            PointerAddress.from_primitive({})
+
+
+class AddressTest(TestCase):
+    def test_from_primitive_invalid_value(self):
+        with self.assertRaises(DeserializeException):
+            Address.from_primitive(1)
+
+        with self.assertRaises(DeserializeException):
+            Address.from_primitive([])
+
+        with self.assertRaises(DeserializeException):
+            Address.from_primitive({})