tests: more e2e tests

Author: zyzniewski-reef

tests/e2e_tests/test_blocks.py

@@ -0,0 +1,22 @@
+import re
+
+
+def test_blocks(subtensor):
+    """
+    Tests:
+    - Get current block
+    - Get block hash
+    - Wait for block
+    """
+
+    block = subtensor.get_current_block()
+
+    assert block == subtensor.block
+
+    block_hash = subtensor.get_block_hash(block)
+
+    assert re.match("0x[a-z0-9]{64}", block_hash)
+
+    subtensor.wait_for_block(block + 10)
+
+    assert subtensor.get_current_block() == block + 10

tests/e2e_tests/test_commit_reveal_v3.py

@@ -53,9 +53,7 @@ async def test_commit_and_reveal_weights_cr3(local_chain, subtensor, alice_walle
     ), "Unable to enable commit reveal on the subnet"
 
     # Verify commit_reveal was enabled
-    assert subtensor.get_subnet_hyperparameters(
-        netuid=netuid,
-    ).commit_reveal_weights_enabled, "Failed to enable commit/reveal"
+    assert subtensor.commit_reveal_enabled(netuid), "Failed to enable commit/reveal"
     logging.console.info("Commit reveal enabled")
 
     # Change the weights rate limit on the subnet

tests/e2e_tests/test_commit_weights.py

@@ -45,9 +45,7 @@ async def test_commit_and_reveal_weights_legacy(local_chain, subtensor, alice_wa
         netuid,
     ), "Unable to enable commit reveal on the subnet"
 
-    assert subtensor.get_subnet_hyperparameters(
-        netuid=netuid,
-    ).commit_reveal_weights_enabled, "Failed to enable commit/reveal"
+    assert subtensor.commit_reveal_enabled(netuid), "Failed to enable commit/reveal"
 
     assert (
         subtensor.get_subnet_hyperparameters(netuid=netuid).commit_reveal_period == 1
@@ -114,11 +112,9 @@ async def test_commit_and_reveal_weights_legacy(local_chain, subtensor, alice_wa
     assert commit_block > 0, f"Invalid block number: {commit_block}"
 
     # Query the WeightCommitRevealInterval storage map
-    reveal_periods = subtensor.query_module(
-        module="SubtensorModule", name="RevealPeriodEpochs", params=[netuid]
-    )
-    periods = reveal_periods
-    assert periods > 0, "Invalid RevealPeriodEpochs"
+    assert (
+        subtensor.get_subnet_reveal_period_epochs(netuid) > 0
+    ), "Invalid RevealPeriodEpochs"
 
     # Wait until the reveal block range
     await wait_epoch(subtensor, netuid)
@@ -187,9 +183,7 @@ async def test_commit_weights_uses_next_nonce(local_chain, subtensor, alice_wall
         netuid,
     ), "Unable to enable commit reveal on the subnet"
 
-    assert subtensor.get_subnet_hyperparameters(
-        netuid=netuid,
-    ).commit_reveal_weights_enabled, "Failed to enable commit/reveal"
+    assert subtensor.commit_reveal_enabled(netuid), "Failed to enable commit/reveal"
 
     assert (
         subtensor.get_subnet_hyperparameters(netuid=netuid).commit_reveal_period == 1