algorand-python-testing 0.4.1__tar.gz → 0.5.0b1__tar.gz

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/.github/workflows/cd.yaml

@@ -45,7 +45,7 @@ jobs:
       - uses: actions/checkout@v4
         with:
           fetch-depth: 0
-          token: ${{ secrets.RELEASE_GITHUB_TOKEN }}
+          token: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Install hatch
         run: pipx install hatch
@@ -79,7 +79,7 @@ jobs:
         if: ${{ github.ref == 'refs/heads/main' }}
         uses: python-semantic-release/python-semantic-release@master
         with:
-          github_token: ${{ secrets.RELEASE_GITHUB_TOKEN }}
+          github_token: ${{ secrets.GITHUB_TOKEN }}
           prerelease: ${{ env.PRERELEASE == 'true' }}
           root_options: $DRY_RUN
 

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/.gitignore

@@ -17,3 +17,4 @@ coverage.xml
 .venv*
 
 .cursorignore
+*.puya.map

algorand_python_testing-0.5.0b1/CHANGELOG.md

@@ -0,0 +1,96 @@
+# CHANGELOG
+## v0.5.0-beta.1 (2025-02-19)
+
+### Documentation
+
+* fix doctest example for `algopy.EllipticCurve` ([`7d0bb0d`](https://github.com/algorandfoundation/algorand-python-testing/commit/7d0bb0dfe9a5ea2d67b130fa300fb80cef52fda5))
+
+## v0.4.1 (2024-09-03)
+
+## v0.4.1-beta.1 (2024-09-03)
+
+## v0.4.0 (2024-08-27)
+
+## v0.4.0-beta.2 (2024-08-27)
+
+## v0.4.0-beta.1 (2024-08-27)
+
+### Documentation
+
+* include usage of `algopy_testing_context` in README.md quick start ([`4702f60`](https://github.com/algorandfoundation/algorand-python-testing/commit/4702f60cfe7d09956a5ae6dbdcd72da29fdda808))
+
+## v0.3.0 (2024-08-22)
+
+## v0.3.0-beta.10 (2024-08-22)
+
+## v0.3.0-beta.9 (2024-08-22)
+
+## v0.3.0-beta.8 (2024-08-22)
+
+## v0.3.0-beta.7 (2024-08-21)
+
+## v0.3.0-beta.6 (2024-08-21)
+
+### Documentation
+
+* documentation for initial stable release of `algorand-python-testing` (#8) ([`9d97d0d`](https://github.com/algorandfoundation/algorand-python-testing/commit/9d97d0de5ff9897e642ec3f11a186f2fb95375bb))
+
+  * docs: wip
+
+  * chore: refresh pyproject
+
+  * docs: refining docs (wip)
+
+  * chore: integrating doctests
+
+  * docs: revamping docs with latest features
+
+  * docs: minor consistency with main readme; patching doctests
+
+  * docs: removing the box from examples
+
+  * docs: refine op codes section
+
+  * chore: merge conflicts
+
+  * chore: apply suggestions from code review
+
+  Co-authored-by: Daniel McGregor <daniel.mcgregor@makerx.com.au>
+
+  * docs: addressing docs pr comments
+
+## v0.3.0-beta.5 (2024-08-21)
+
+## v0.3.0-beta.4 (2024-08-21)
+
+### Documentation
+
+* integrating pydoclint; formatting docs; removing docs from stub implementation ([`d729bf9`](https://github.com/algorandfoundation/algorand-python-testing/commit/d729bf9b70ef885cd1b2ef705c4f5e2582d853ab))
+
+## v0.3.0-beta.3 (2024-08-16)
+
+## v0.3.0-beta.2 (2024-08-16)
+
+## v0.3.0-beta.1 (2024-08-14)
+
+## v0.2.2-beta.5 (2024-07-30)
+
+### Documentation
+
+* patch urls in README.md (#9) ([`eddf612`](https://github.com/algorandfoundation/algorand-python-testing/commit/eddf612b177a2acddf15d58be3f375e99fb6564b))
+
+  * chore: patch urls in README.md
+
+  * ci: relaxing rules around paths-ignore
+
+  * docs: patching old namespace name in readme
+
+## v0.2.2-beta.4 (2024-07-25)
+
+## v0.2.2-beta.3 (2024-07-25)
+
+## v0.2.2-beta.2 (2024-07-25)
+
+## v0.2.2-beta.1 (2024-07-24)
+
+## v0.2.1 (2024-07-10)

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: algorand-python-testing
-Version: 0.4.1
+Version: 0.5.0b1
 Summary: Algorand Python testing library
 Project-URL: Documentation, https://github.com/algorandfoundation/puya/tree/main/algopy_testing#README.md
 Project-URL: Issues, https://github.com/algorandfoundation/puya/issues
@@ -48,7 +48,7 @@ Alternatively, if you want to start from scratch:
 2. Install [AlgoKit CLI](https://github.com/algorandfoundation/algokit-cli?tab=readme-ov-file#install)
 3. Install Algorand Python Testing into your project:
     ```bash
-    pip install algorand-testing-python
+    pip install algorand-python-testing
     ```
 4. Create a test file (e.g., `test_contract.py`):
 

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/README.md

@@ -25,7 +25,7 @@ Alternatively, if you want to start from scratch:
 2. Install [AlgoKit CLI](https://github.com/algorandfoundation/algokit-cli?tab=readme-ov-file#install)
 3. Install Algorand Python Testing into your project:
     ```bash
-    pip install algorand-testing-python
+    pip install algorand-python-testing
     ```
 4. Create a test file (e.g., `test_contract.py`):
 

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/docs/testing-guide/opcodes.md

@@ -353,20 +353,22 @@ test_mock_vrf_verify()
 from unittest.mock import patch, MagicMock
 import algopy
 
-def test_mock_elliptic_curve_decompress():
-    mock_result = (algopy.Bytes(b'x_coord'), algopy.Bytes(b'y_coord'))
-    with patch('algopy.op.EllipticCurve.decompress', return_value=mock_result) as mock_decompress:
-        result = algopy.op.EllipticCurve.decompress(
+def test_mock_elliptic_curve_add():
+    mock_result = algopy.Bytes(b'result')
+    with patch('algopy.op.EllipticCurve.add', return_value=mock_result) as mock_add:
+        result = algopy.op.EllipticCurve.add(
             algopy.op.EC.BN254g1,
-            algopy.Bytes(b'compressed_point')
+            algopy.Bytes(b'a'),
+            algopy.Bytes(b'b')
         )
     assert result == mock_result
-    mock_decompress.assert_called_once_with(
+    mock_add.assert_called_once_with(
         algopy.op.EC.BN254g1,
-        algopy.Bytes(b'compressed_point')
+        algopy.Bytes(b'a'),
+        algopy.Bytes(b'b'),
     )
 
-test_mock_elliptic_curve_decompress()
+test_mock_elliptic_curve_add()
 ```
 
 These examples demonstrate how to mock key mockable opcodes in `algorand-python-testing`. Use similar techniques (in your preferred testing framework) for other mockable opcodes like `algopy.compile_logicsig`, `algopy.arc4.arc4_create`, and `algopy.arc4.arc4_update`.

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "algorand-python-testing"
-version = "0.4.1"
+version = "0.5.0-beta.1"
 description = 'Algorand Python testing library'
 readme = "README.md"
 requires-python = ">=3.12"
@@ -57,7 +57,7 @@ dependencies = [
     "pytest-mock>=3.10.0",
     "pytest-xdist[psutil]>=3.3",
     "py-algorand-sdk>=2.4.0",
-    "algokit-utils>=2.2.1",
+    "algokit-utils>=3.0.0",
     "pytest-cov>=4.1.0",
     "prettytable>=3.9.0",
     "mypy==1.10",
@@ -171,7 +171,7 @@ dependencies = [
     "pytest-mock>=3.10.0",
     "pytest-xdist[psutil]>=3.3",
     "py-algorand-sdk>=2.4.0",
-    "algokit-utils>=2.2.1",
+    "algokit-utils>=3.0.0",
     "pytest-cov>=4.1.0",
     "mypy==1.10",
 ]

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/scripts/check_stubs_cov.py

@@ -1,7 +1,9 @@
 import ast
+import importlib
 import inspect
 import site
 import sys
+import typing
 from collections.abc import Iterable
 from pathlib import Path
 from typing import NamedTuple
@@ -13,6 +15,40 @@ SITE_PACKAGES = Path(site.getsitepackages()[0])
 STUBS_ROOT = SITE_PACKAGES / "algopy-stubs"
 IMPL = PROJECT_ROOT / "src"
 ROOT_MODULE = "algopy"
+_ADDITIONAL_GLOBAL_IMPLS = [
+    "_algopy_testing.op.global_values._Global",
+    "_algopy_testing.op.global_values.GlobalFields",
+]
+_ADDITIONAL_TXN_IMPLS = [
+    "_algopy_testing.models.txn_fields.TransactionFields",
+    "_algopy_testing.models.txn_fields.TransactionFieldsGetter",
+    "_algopy_testing.op.constants.OP_MEMBER_TO_TXN_MEMBER",
+]
+
+# mapping of stub types to additional implementation types to scan for members
+_ADDITIONAL_TYPE_IMPLS = {
+    "algopy.Asset": ["_algopy_testing.models.asset.AssetFields"],
+    "algopy.Account": ["_algopy_testing.models.account.AccountFields"],
+    "algopy.Application": ["_algopy_testing.models.application.ApplicationFields"],
+    "algopy.Global": _ADDITIONAL_GLOBAL_IMPLS,
+    "algopy.Txn": _ADDITIONAL_TXN_IMPLS,
+    "algopy.op.Global": _ADDITIONAL_GLOBAL_IMPLS,
+    "algopy.op.GTxn": _ADDITIONAL_TXN_IMPLS,
+    "algopy.op.GITxn": _ADDITIONAL_TXN_IMPLS,
+    "algopy.op.Txn": _ADDITIONAL_TXN_IMPLS,
+    "algopy.op.ITxn": _ADDITIONAL_TXN_IMPLS,
+    "algopy.op.ITxnCreate": _ADDITIONAL_TXN_IMPLS,
+    "algopy.op.AppParamsGet": ["_algopy_testing.op.misc._AppParamsGet"],
+    "algopy.op.AssetHoldingGet": ["_algopy_testing.op.misc._AssetHoldingGet"],
+    "algopy.op.AppGlobal": ["_algopy_testing.op.misc._AppGlobal"],
+    "algopy.op.AppLocal": ["_algopy_testing.op.misc._AppLocal"],
+    "algopy.op.Scratch": ["_algopy_testing.op.misc._Scratch"],
+}
+
+# mapping of stub types to members that may be present but not found when discovering members
+_ADDITIONAL_MEMBERS = {
+    "algopy.Asset": ["id"],
+}
 
 
 class ASTNodeDefinition(NamedTuple):
@@ -127,9 +163,6 @@ def collect_stubs(stubs_dir: Path, relative_module: str) -> dict[str, ASTNodeDef
 def collect_coverage(stubs: dict[str, ASTNodeDefinition]) -> list[CoverageResult]:
     result = []
     for full_name, stub in stubs.items():
-        if "GTxn" in full_name:
-            print("stop")
-
         coverage = _get_impl_coverage(full_name, stub)
         if coverage:
             try:
@@ -145,7 +178,7 @@ def collect_coverage(stubs: dict[str, ASTNodeDefinition]) -> list[CoverageResult
             CoverageResult(
                 full_name=full_name,
                 stub_file=str(stub.path.relative_to(STUBS_ROOT)),
-                impl_file=impl_file,
+                impl_file=impl_file or "MISSING!",
                 coverage=coverage.coverage if coverage else 0,
                 missing=", ".join(coverage.missing if coverage else []),
             )
@@ -156,18 +189,16 @@ def collect_coverage(stubs: dict[str, ASTNodeDefinition]) -> list[CoverageResult
 def print_results(results: list[CoverageResult]) -> None:
     table = PrettyTable(
         field_names=["Name", "Implementation", "Source Stub", "Coverage", "Missing"],
-        sortby="Coverage",
         header=True,
         border=True,
         padding_width=2,
-        reversesort=True,
         left_padding_width=0,
         right_padding_width=1,
         align="l",
         max_width=100,
     )
 
-    for result in results:
+    for result in sorted(results, key=lambda c: c.coverage):
         table.add_row(
             [
                 result.full_name,
@@ -209,7 +240,6 @@ def _get_impl_coverage(symbol: str, stub: ASTNodeDefinition) -> ImplCoverage | N
     try:
         impl = getattr(mod, name)
     except AttributeError:
-        print(f"Attribute {name} not found in module {module}")
         return None
 
     try:
@@ -219,20 +249,6 @@ def _get_impl_coverage(symbol: str, stub: ASTNodeDefinition) -> ImplCoverage | N
         if hasattr(impl, "__class__"):
             try:
                 impl_path = Path(inspect.getfile(impl.__class__))
-                # For special cases like GTxn and GITxn, assume full implementation
-                if name in [
-                    "GTxn",
-                    "GITxn",
-                    "Txn",
-                    "ITxn",
-                    "Global",
-                    "AssetConfigInnerTransaction",
-                    "Contract",
-                    "ApplicationCallInnerTransaction",
-                    "UFixedNxM",
-                    "BigUFixedNxM",
-                ]:
-                    return ImplCoverage(impl_path)
             except TypeError:
                 print(f"Warning: Could not determine file for {symbol}")
                 return None
@@ -240,18 +256,56 @@ def _get_impl_coverage(symbol: str, stub: ASTNodeDefinition) -> ImplCoverage | N
             print(f"Warning: Could not determine file for {symbol}")
             return None
 
-    return _compare_stub_impl(stub.node, impl, impl_path)
+    return _compare_stub_impl(stub.node, symbol, impl, impl_path)
+
+
+def _get_impl_members(impl_name: str, impl: object) -> set[str]:
+    if isinstance(impl, type):
+        impl_mros: list[object] = [
+            typ for typ in impl.mro() if typ.__module__.startswith("_algopy_testing")
+        ]
+    else:
+        impl_mros = []
+    for additional_type in _ADDITIONAL_TYPE_IMPLS.get(impl_name, []):
+        impl_mros.append(_resolve_fullname(additional_type))
+
+    impl_members = set[str](_ADDITIONAL_MEMBERS.get(impl_name, []))
+    for impl_typ in impl_mros:
+        if typing.is_typeddict(impl_typ) and isinstance(impl_typ, type):
+            for typed_dict_mro in impl_typ.mro():
+                ann = getattr(typed_dict_mro, "__annotations__", None)
+                if isinstance(ann, dict):
+                    impl_members.update(ann.keys())
+        elif isinstance(impl_typ, dict):
+            impl_members.update(impl_typ.keys())
+        elif isinstance(impl_typ, type):
+            members = list(vars(impl_typ).keys())
+            impl_members.update(members)
+        else:
+            raise ValueError(f"unexpected implementation type, {impl_typ}")
+    # special case for ITxnCreate
+    if impl_name == "algopy.op.ITxnCreate":
+        impl_members = {f"set_{member}" for member in impl_members}
+        impl_members.update(("begin", "next", "submit"))
+    return impl_members
+
 
+def _resolve_fullname(fullname: str) -> object:
+    # note this assumes no nested classes
+    module_name, type_name = fullname.rsplit(".", maxsplit=1)
+    module = importlib.import_module(module_name)
+    return getattr(module, type_name)
 
-def _compare_stub_impl(stub: ast.AST, impl: object, impl_path: Path) -> ImplCoverage:
+
+def _compare_stub_impl(
+    stub: ast.AST, impl_name: str, impl: object, impl_path: Path
+) -> ImplCoverage:
     # classes are really the only types that can be "partially implemented"
     # from a typing perspective
-    if not isinstance(stub, ast.ClassDef):
+    # algopy.uenumerate is typed as a class, but is really just a function
+    if not isinstance(stub, ast.ClassDef) or impl_name == "algopy.uenumerate":
         return ImplCoverage(impl_path)
-
-    # using vars to only get explicitly implemented members
-    # need more sophisticated approach if implementations start using inheritance
-    impl_members = set(vars(impl))
+    impl_members = _get_impl_members(impl_name, impl)
     stub_members = set()
     for stmt in stub.body:
         if isinstance(stmt, ast.FunctionDef):
@@ -287,6 +341,8 @@ def _compare_stub_impl(stub: ast.AST, impl: object, impl_path: Path) -> ImplCove
             "ne",
         )
     }
+    # excluding special fields used in typing hints
+    default_impls.update(("__match_args__", "__match_value__"))
     missing = sorted(stub_members.difference({*impl_members, *default_impls}))
     return ImplCoverage(impl_path, sorted(stub_members), missing)
 

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/scripts/refresh_test_artifacts.py

@@ -16,13 +16,13 @@ ENV_WITH_NO_COLOR = dict(os.environ) | {
 
 def get_artifact_folders(root_dir: str) -> Iterator[Path]:
     for folder in Path(root_dir).iterdir():
-        if folder.is_dir() and (folder / "contract.py").exists():
+        if folder.is_dir() and not str(folder.stem).startswith((".", "__")):
             yield folder
 
 
 def compile_contract(folder: Path) -> None:
     logger.info(f"Compiling: {folder}")
-    contract_path = folder / "contract.py"
+    contract_path = folder
     (folder / "data").mkdir(exist_ok=True)
     compile_cmd = [
         "hatch",

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/src/_algopy_testing/arc4.py

@@ -17,6 +17,7 @@ from _algopy_testing.constants import (
     UINT512_SIZE,
 )
 from _algopy_testing.models.account import Account
+from _algopy_testing.models.contract import ARC4Contract
 from _algopy_testing.mutable import (
     MutableBytes,
     add_mutable_callback,
@@ -42,6 +43,7 @@ if typing.TYPE_CHECKING:
 
 __all__ = [
     "ARC4Client",
+    "ARC4Contract",
     "Address",
     "BigUFixedNxM",
     "BigUIntN",
@@ -743,7 +745,7 @@ class _DynamicArrayTypeInfo(_TypeInfo):
         return True
 
 
-class _DynamicArrayMeta(type(_ABIEncoded), typing.Generic[_TArrayItem, _TArrayLength]):  # type: ignore  # noqa: PGH003
+class _DynamicArrayMeta(type(_ABIEncoded), typing.Generic[_TArrayItem]):  # type: ignore[misc]
     __concrete__: typing.ClassVar[dict[type, type]] = {}
 
     def __getitem__(cls, key_t: type[_TArrayItem]) -> type:
@@ -1013,17 +1015,18 @@ class Tuple(
 
 
 class _StructTypeInfo(_TypeInfo):
-    def __init__(self, struct_type: type[Struct]) -> None:
+    def __init__(self, struct_type: type[Struct], *, frozen: bool) -> None:
         self.struct_type = struct_type
         self.fields = dataclasses.fields(struct_type)
         self.field_names = [field.name for field in self.fields]
+        self.frozen = frozen
 
     @property
     def typ(self) -> type:
         return self.struct_type
 
     @property
-    def child_types(self) -> Iterable[_TypeInfo]:
+    def child_types(self) -> list[_TypeInfo]:
         return _tuple_type_from_struct(self.struct_type)._type_info.child_types
 
     @property
@@ -1054,8 +1057,11 @@ class Struct(MutableBytes, _ABIEncoded, metaclass=_StructMeta):  # type: ignore[
     _type_info: typing.ClassVar[_StructTypeInfo]  # type: ignore[misc]
 
     def __init_subclass__(cls, *args: typing.Any, **kwargs: dict[str, typing.Any]) -> None:
-        dataclasses.dataclass(cls, *args, **kwargs)
-        cls._type_info = _StructTypeInfo(cls)
+        # make implementation not frozen, so we can conditionally control behaviour
+        dataclasses.dataclass(cls, *args, **{**kwargs, "frozen": False})
+        frozen = kwargs.get("frozen", False)
+        assert isinstance(frozen, bool)
+        cls._type_info = _StructTypeInfo(cls, frozen=frozen)
 
     def __post_init__(self) -> None:
         # calling base class here to init Mutable
@@ -1071,6 +1077,10 @@ class Struct(MutableBytes, _ABIEncoded, metaclass=_StructMeta):  # type: ignore[
         super().__setattr__(key, value)
         # don't update backing value until base class has been init'd
         if hasattr(self, "_on_mutate") and key in self._type_info.field_names:
+            if self._type_info.frozen:
+                raise dataclasses.FrozenInstanceError(
+                    f"{type(self)} is frozen and cannot be modified"
+                )
             self._update_backing_value()
 
     def _update_backing_value(self) -> None:
@@ -1094,6 +1104,12 @@ class Struct(MutableBytes, _ABIEncoded, metaclass=_StructMeta):  # type: ignore[
         tuple_items = tuple(getattr(self, field.name) for field in dataclasses.fields(self))
         return Tuple(tuple_items)
 
+    def _replace(self, **kwargs: typing.Any) -> typing.Self:
+        copy = self.copy()
+        for field, value in kwargs.items():
+            setattr(copy, field, value)
+        return copy
+
 
 class ARC4Client:
     pass
@@ -1146,34 +1162,16 @@ def emit(event: str | Struct, /, *args: object) -> None:
     log(event_hash[:4] + event_data.value)
 
 
-def native_value_to_arc4(value: object) -> _ABIEncoded:  # noqa: PLR0911
-    import algopy
-
-    if isinstance(value, _ABIEncoded):
-        return value
-    if isinstance(value, bool):
-        return Bool(value)
-    if isinstance(value, algopy.UInt64):
-        return UInt64(value)
-    if isinstance(value, algopy.BigUInt):
-        return UInt512(value)
-    if isinstance(value, algopy.Bytes):
-        return DynamicBytes(value)
-    if isinstance(value, algopy.String):
-        return String(value)
-    if isinstance(value, tuple):
-        return Tuple(tuple(map(native_value_to_arc4, value)))
-    raise TypeError(f"Unsupported type: {type(value).__name__}")
-
-
 def _cast_arg_as_arc4(arg: object) -> _ABIEncoded:
+    from _algopy_testing.serialize import native_to_arc4
+
     if isinstance(arg, int) and not isinstance(arg, bool):
         return UInt64(arg) if arg <= MAX_UINT64 else UInt512(arg)
     if isinstance(arg, bytes):
         return DynamicBytes(arg)
     if isinstance(arg, str):
         return String(arg)
-    return native_value_to_arc4(arg)
+    return native_to_arc4(arg)
 
 
 def _find_bool(
@@ -1237,13 +1235,13 @@ def _get_max_bytes_len(type_info: _TypeInfo) -> int:
     size = 0
     if isinstance(type_info, _DynamicArrayTypeInfo):
         size += _ABI_LENGTH_SIZE
-    elif isinstance(type_info, _TupleTypeInfo | _StaticArrayTypeInfo):
+    elif isinstance(type_info, _TupleTypeInfo | _StructTypeInfo | _StaticArrayTypeInfo):
         i = 0
-        child_types = (
-            type_info.child_types
-            if isinstance(type_info, _TupleTypeInfo)
-            else [type_info.item_type] * type_info.size
-        )
+        if isinstance(type_info, _TupleTypeInfo | _StructTypeInfo):
+            child_types = type_info.child_types
+        else:
+            typing.assert_type(type_info, _StaticArrayTypeInfo)
+            child_types = [type_info.item_type] * type_info.size
         while i < len(child_types):
             if isinstance(child_types[i], _BoolTypeInfo):
                 after = _find_bool_types(child_types, i, 1)

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/src/_algopy_testing/compiled.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 # ruff: noqa: ARG001, PLR0913
-import dataclasses
 import typing
 
 from _algopy_testing.utils import raise_mocked_function_error
@@ -12,8 +11,7 @@ if typing.TYPE_CHECKING:
     import algopy
 
 
-@dataclasses.dataclass
-class CompiledContract:
+class CompiledContract(typing.NamedTuple):
     approval_program: tuple[algopy.Bytes, algopy.Bytes]
     clear_state_program: tuple[algopy.Bytes, algopy.Bytes]
     extra_program_pages: algopy.UInt64
@@ -23,8 +21,7 @@ class CompiledContract:
     local_bytes: algopy.UInt64
 
 
-@dataclasses.dataclass
-class CompiledLogicSig:
+class CompiledLogicSig(typing.NamedTuple):
     account: algopy.Account
 
 

{algorand_python_testing-0.4.1 → algorand_python_testing-0.5.0b1}/src/_algopy_testing/context_helpers/ledger_context.py

@@ -3,6 +3,8 @@ from __future__ import annotations
 import typing
 from collections import defaultdict
 
+import algosdk.constants
+
 from _algopy_testing.constants import MAX_BOX_SIZE
 from _algopy_testing.models.account import Account
 from _algopy_testing.primitives.uint64 import UInt64
@@ -31,7 +33,7 @@ class LedgerContext:
         self._account_data = defaultdict[str, AccountContextData](get_empty_account)
         self._app_data: dict[int, ApplicationContextData] = {}
         self._asset_data: dict[int, AssetFields] = {}
-        self._blocks: dict[int, dict[str, int]] = {}
+        self._blocks: dict[int, dict[str, int | bytes | str]] = {}
         self._global_fields: GlobalFields = get_default_global_fields()
 
         self._asset_id = iter(range(1001, 2**64))
@@ -365,8 +367,19 @@ class LedgerContext:
         boxes = self._get_app_data(app).boxes
         return _as_box_key(key) in boxes
 
-    def set_block(
-        self, index: int, seed: algopy.UInt64 | int, timestamp: algopy.UInt64 | int
+    def set_block(  # noqa: PLR0913
+        self,
+        index: int,
+        seed: algopy.UInt64 | int,
+        timestamp: algopy.UInt64 | int,
+        bonus: algopy.UInt64 | int = 0,
+        branch: algopy.Bytes | bytes = b"",
+        fee_sink: algopy.Account | str = algosdk.constants.ZERO_ADDRESS,
+        fees_collected: algopy.UInt64 | int = 0,
+        proposer: algopy.Account | str = algosdk.constants.ZERO_ADDRESS,
+        proposer_payout: algopy.UInt64 | int = 0,
+        protocol: algopy.Bytes | bytes = b"",
+        txn_counter: algopy.UInt64 | int = 0,
     ) -> None:
         """Set block content.
 
@@ -374,10 +387,29 @@ class LedgerContext:
             index (int): The block index.
             seed (algopy.UInt64 | int): The block seed.
             timestamp (algopy.UInt64 | int): The block timestamp.
+            bonus (algopy.UInt64 | int): The block bonus.
+            branch (algopy.Bytes | bytes): The block branch.
+            fee_sink (algopy.Account | str): The block fee sink.
+            fees_collected (algopy.UInt64 | int): The fess collected.
+            proposer (algopy.Account | str): The block proposer.
+            proposer_payout (algopy.UInt64 | int): The block proposer payout.
+            protocol (algopy.Bytes | bytes): The block protocol.
+            txn_counter (algopy.UInt64 | int): The block transaction counter.
         """
-        self._blocks[index] = {"seed": int(seed), "timestamp": int(timestamp)}
-
-    def get_block_content(self, index: int, key: str) -> int:
+        self._blocks[index] = {
+            "seed": int(seed),
+            "timestamp": int(timestamp),
+            "bonus": int(bonus),
+            "branch": as_bytes(branch),
+            "fee_sink": str(fee_sink),
+            "fees_collected": int(fees_collected),
+            "proposer": str(proposer),
+            "proposer_payout": int(proposer_payout),
+            "protocol": as_bytes(protocol),
+            "txn_counter": int(txn_counter),
+        }
+
+    def get_block_content(self, index: int, key: str) -> int | bytes | str:
         """Get block content.
 
         Args: