testcase

algorand · Sep 22, 2022 · 6acab46 · 6acab46
1 parent 13a00ad
commit 6acab46
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 0 deletions.
diff --git a/pyteal/ast/abi/__init__.py b/pyteal/ast/abi/__init__.py
@@ -84,6 +84,7 @@
     type_spec_from_annotation,
     type_specs_from_signature,
     contains_type_spec,
+    type_spec_is_assignable,
 )
 
 __all__ = [
@@ -167,4 +168,5 @@
     "algosdk_from_annotation",
     "algosdk_from_type_spec",
     "contains_type_spec",
+    "type_spec_is_assignable",
 ]
diff --git a/pyteal/ast/abi/util_test.py b/pyteal/ast/abi/util_test.py
@@ -771,3 +771,79 @@ def test_sdk_type_specs_from_signature(sig_str, sig_args, sig_rets):
     args, ret = abi.type_specs_from_signature(sig_str)
     assert args == sig_args
     assert ret == sig_rets
+
+
+class NamedTDecl(abi.NamedTuple):
+    a: abi.Field[abi.Uint64]
+    b: abi.Field[
+        abi.Tuple3[abi.PaymentTransaction, abi.Address, abi.StaticBytes[Literal[16]]]
+    ]
+    c: abi.Field[abi.Transaction]
+
+
+TYPE_SPEC_ASSIGNABLE_CASES = [
+    (abi.PaymentTransactionTypeSpec(), abi.TransactionTypeSpec(), True),
+    (
+        abi.type_spec_from_annotation(abi.StaticArray[abi.Byte, Literal[10]]),
+        abi.type_spec_from_annotation(abi.StaticBytes[Literal[10]]),
+        True,
+    ),
+    (
+        abi.type_spec_from_annotation(abi.StaticBytes[Literal[10]]),
+        abi.type_spec_from_annotation(abi.StaticArray[abi.Byte, Literal[10]]),
+        True,
+    ),
+    (
+        abi.type_spec_from_annotation(abi.DynamicBytes),
+        abi.type_spec_from_annotation(abi.DynamicArray[abi.Byte]),
+        True,
+    ),
+    (
+        abi.type_spec_from_annotation(abi.DynamicArray[abi.Byte]),
+        abi.type_spec_from_annotation(abi.DynamicBytes),
+        True,
+    ),
+    (
+        abi.type_spec_from_annotation(abi.DynamicArray[abi.Uint8]),
+        abi.type_spec_from_annotation(abi.DynamicBytes),
+        False,
+    ),
+    (
+        abi.type_spec_from_annotation(abi.DynamicBytes),
+        abi.type_spec_from_annotation(abi.DynamicArray[abi.Uint8]),
+        False,
+    ),
+    (
+        abi.type_spec_from_annotation(NamedTDecl),
+        abi.type_spec_from_annotation(
+            abi.Tuple3[
+                abi.Uint64,
+                abi.Tuple3[
+                    abi.Transaction, abi.Address, abi.StaticArray[abi.Byte, Literal[16]]
+                ],
+                abi.Transaction,
+            ]
+        ),
+        True,
+    ),
+    (
+        abi.type_spec_from_annotation(
+            abi.Tuple3[
+                abi.Uint64,
+                abi.Tuple3[
+                    abi.PaymentTransaction,
+                    abi.Address,
+                    abi.StaticArray[abi.Byte, Literal[16]],
+                ],
+                abi.Transaction,
+            ]
+        ),
+        abi.type_spec_from_annotation(NamedTDecl),
+        True,
+    ),
+]
+
+
+@pytest.mark.parametrize("a, b, expected", TYPE_SPEC_ASSIGNABLE_CASES)
+def test_type_spec_assignable(a, b, expected):
+    assert abi.type_spec_is_assignable(a, b) == expected