Use assert_type alongside cast

setup.cfg

@@ -113,7 +113,7 @@ python_requires = >=3.10
 install_requires =
     pycryptodome>=3,<4
     coincurve>=20,<21
-    typing_extensions>=4
+    typing_extensions>=4.2
     py_ecc @ git+https://github.com/petertdavies/py_ecc.git@127184f4c57b1812da959586d0fe8f43bb1a2389
     ethereum-types>=0.2.1,<0.3
 

src/ethereum/arrow_glacier/trie.py

@@ -33,6 +33,7 @@
 from ethereum_types.bytes import Bytes
 from ethereum_types.frozen import slotted_freezable
 from ethereum_types.numeric import U256, Uint
+from typing_extensions import assert_type
 
 from ethereum.crypto.hash import keccak256
 from ethereum.london import trie as previous_trie
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/berlin/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/byzantium/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/cancun/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -488,6 +489,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/constantinople/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/dao_fork/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/frontier/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -486,6 +487,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/gray_glacier/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/homestead/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/istanbul/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/london/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/muir_glacier/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/paris/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/shanghai/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -488,6 +489,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/spurious_dragon/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )

src/ethereum/tangerine_whistle/trie.py

@@ -27,6 +27,7 @@
     Tuple,
     TypeVar,
     Union,
+    assert_type,
     cast,
 )
 
@@ -485,6 +486,6 @@ def patricialize(
         for k in range(16)
     )
     return BranchNode(
-        cast(BranchSubnodes, subnodes),
+        cast(BranchSubnodes, assert_type(subnodes, Tuple[rlp.Extended, ...])),
         value,
     )