Fix abi-docs tests (#402)

algorand · Jun 17, 2022 · d6e8501 · d6e8501
1 parent e5c3101
commit d6e8501
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/pyteal/ast/abi/util_test.py b/pyteal/ast/abi/util_test.py
@@ -491,61 +491,61 @@ def test_size_of():
     # ),
     (algosdk.abi.TupleType([]), "()", abi.TupleTypeSpec(), abi.Tuple0),
     (
-        "cannot map ABI transaction type spec <pyteal.TransactionTypeSpec",
+        "cannot map ABI transaction type spec <pyteal.abi.TransactionTypeSpec",
         "txn",
         abi.TransactionTypeSpec(),
         abi.Transaction,
     ),
     (
-        "cannot map ABI transaction type spec <pyteal.PaymentTransactionTypeSpec",
+        "cannot map ABI transaction type spec <pyteal.abi.PaymentTransactionTypeSpec",
         "pay",
         abi.PaymentTransactionTypeSpec(),
         abi.PaymentTransaction,
     ),
     (
-        "cannot map ABI transaction type spec <pyteal.KeyRegisterTransactionTypeSpec",
+        "cannot map ABI transaction type spec <pyteal.abi.KeyRegisterTransactionTypeSpec",
         "keyreg",
         abi.KeyRegisterTransactionTypeSpec(),
         abi.KeyRegisterTransaction,
     ),
     (
-        "cannot map ABI transaction type spec <pyteal.AssetConfigTransactionTypeSpec",
+        "cannot map ABI transaction type spec <pyteal.abi.AssetConfigTransactionTypeSpec",
         "acfg",
         abi.AssetConfigTransactionTypeSpec(),
         abi.AssetConfigTransaction,
     ),
     (
-        "cannot map ABI transaction type spec <pyteal.AssetTransferTransactionTypeSpec",
+        "cannot map ABI transaction type spec <pyteal.abi.AssetTransferTransactionTypeSpec",
         "axfer",
         abi.AssetTransferTransactionTypeSpec(),
         abi.AssetTransferTransaction,
     ),
     (
-        "cannot map ABI transaction type spec <pyteal.AssetFreezeTransactionTypeSpec",
+        "cannot map ABI transaction type spec <pyteal.abi.AssetFreezeTransactionTypeSpec",
         "afrz",
         abi.AssetFreezeTransactionTypeSpec(),
         abi.AssetFreezeTransaction,
     ),
     (
-        "cannot map ABI transaction type spec <pyteal.ApplicationCallTransactionTypeSpec",
+        "cannot map ABI transaction type spec <pyteal.abi.ApplicationCallTransactionTypeSpec",
         "appl",
         abi.ApplicationCallTransactionTypeSpec(),
         abi.ApplicationCallTransaction,
     ),
     (
-        "cannot map ABI reference type spec <pyteal.AccountTypeSpec",
+        "cannot map ABI reference type spec <pyteal.abi.AccountTypeSpec",
         "account",
         abi.AccountTypeSpec(),
         abi.Account,
     ),
     (
-        "cannot map ABI reference type spec <pyteal.ApplicationTypeSpec",
+        "cannot map ABI reference type spec <pyteal.abi.ApplicationTypeSpec",
         "application",
         abi.ApplicationTypeSpec(),
         abi.Application,
     ),
     (
-        "cannot map ABI reference type spec <pyteal.AssetTypeSpec",
+        "cannot map ABI reference type spec <pyteal.abi.AssetTypeSpec",
         "asset",
         abi.AssetTypeSpec(),
         abi.Asset,

diff --git a/pyteal/ast/subroutine_test.py b/pyteal/ast/subroutine_test.py
@@ -1023,15 +1023,15 @@ def fnWithMixedAnnsABIRet2(
         ),
         (
             fnWithMixedAnnsABIRet1,
-            "Function has return of disallowed type pyteal.StaticArray[pyteal.Uint32, typing.Literal[10]]. "
+            "Function has return of disallowed type pyteal.abi.StaticArray[pyteal.abi.Uint32, typing.Literal[10]]. "
             "Only Expr is allowed",
-            "Function has return of disallowed type pyteal.StaticArray[pyteal.Uint32, typing.Literal[10]]. "
+            "Function has return of disallowed type pyteal.abi.StaticArray[pyteal.abi.Uint32, typing.Literal[10]]. "
             "Only Expr is allowed",
         ),
         (
             fnWithMixedAnnsABIRet2,
-            "Function has return of disallowed type <class 'pyteal.Uint64'>. Only Expr is allowed",
-            "Function has return of disallowed type <class 'pyteal.Uint64'>. Only Expr is allowed",
+            "Function has return of disallowed type <class 'pyteal.abi.Uint64'>. Only Expr is allowed",
+            "Function has return of disallowed type <class 'pyteal.abi.Uint64'>. Only Expr is allowed",
         ),
     )