removed manual fork checks in favor of abstracted, dynamic checks

Author: felix314159

src/ethereum_test_types/blob.py

@@ -263,8 +263,11 @@ def write_to_file(self):
 
     def verify_cell_kzg_proof_batch(self, cell_indices: list) -> bool:
         """Check whether all cell proofs are valid and returns True only if that is the case."""
-        assert self.fork in ["osaka"], (
-            f"verify_cell_kzg_proof_batch() is not available for fork: {self.fork}"
+        fork_obj = fork_string_to_object(self.fork)
+        amount_cell_proofs: int = cast(int, fork_obj.get_blob_constant("AMOUNT_CELL_PROOFS"))
+
+        assert amount_cell_proofs > 0, (
+            f"verify_cell_kzg_proof_batch() is not available for your fork: {self.fork}."
         )
 
         assert self.cells is not None, "self.cells is None, critical error."
@@ -294,7 +297,10 @@ def delete_cells_then_recover_them(self, deletion_indices: list[int]):
         the ckzg recovery mechanism is used to repair the missing cells.
         If no assertion is triggered the reconstruction was successful.
         """
-        assert self.fork in ["osaka"], (
+        fork_obj = fork_string_to_object(self.fork)
+        amount_cell_proofs: int = cast(int, fork_obj.get_blob_constant("AMOUNT_CELL_PROOFS"))
+
+        assert amount_cell_proofs > 0, (
             f"delete_cells_then_recover_them() is not available for fork: {self.fork}"
         )
 
@@ -370,7 +376,9 @@ def corrupt_byte(b: bytes) -> Bytes:
             return Bytes(bytes([b[0] ^ 0xFF]))
 
         # osaka and later
-        if self.fork in ["osaka"]:
+        fork_obj = fork_string_to_object(self.fork)
+        amount_cell_proofs: int = cast(int, fork_obj.get_blob_constant("AMOUNT_CELL_PROOFS"))
+        if amount_cell_proofs > 0:
             assert isinstance(self.proof, list), (
                 "proof was expected to be a list but it isn't"
             )  # make mypy happy
@@ -396,7 +404,7 @@ def corrupt_byte(b: bytes) -> Bytes:
             return
 
         # pre-osaka (cancun and prague)
-        assert self.fork in ["cancun", "prague"], (
+        assert amount_cell_proofs == 0, (
             f"You need to adjust corrupt_proof to handle fork {self.fork}"
         )
         assert isinstance(self.proof, Bytes), "proof was expected to be Bytes but it isn't"