Merge pull request #2722 from opentensor/fix/test-incentive

Updates test_incentive

Author: ibraheem-opentensor

tests/e2e_tests/test_incentive.py

@@ -2,13 +2,17 @@
 
 import pytest
 
+from bittensor import Balance
+
 from tests.e2e_tests.utils.chain_interactions import (
     sudo_set_hyperparameter_values,
     wait_epoch,
+    sudo_set_admin_utils,
 )
 
 
 @pytest.mark.asyncio
+@pytest.mark.parametrize("local_chain", [False], indirect=True)
 async def test_incentive(local_chain, subtensor, templates, alice_wallet, bob_wallet):
     """
     Test the incentive mechanism and interaction of miners/validators
@@ -31,6 +35,21 @@ async def test_incentive(local_chain, subtensor, templates, alice_wallet, bob_wa
     # Verify subnet <netuid> created successfully
     assert subtensor.subnet_exists(netuid), "Subnet wasn't created successfully"
 
+    # Change tempo to 10
+    tempo_set = 10
+    assert (
+        sudo_set_admin_utils(
+            local_chain,
+            alice_wallet,
+            call_function="sudo_set_tempo",
+            call_params={"netuid": netuid, "tempo": tempo_set},
+            return_error_message=True,
+        )[0]
+        is True
+    )
+    tempo = subtensor.get_subnet_hyperparameters(netuid=netuid).tempo
+    assert tempo_set == tempo
+
     # Register Bob as a neuron on the subnet
     assert subtensor.burned_register(
         bob_wallet, netuid
@@ -41,6 +60,27 @@ async def test_incentive(local_chain, subtensor, templates, alice_wallet, bob_wa
         len(subtensor.neurons(netuid=netuid)) == 2
     ), "Alice & Bob not registered in the subnet"
 
+    # Add stake for Alice
+    assert subtensor.add_stake(
+        alice_wallet,
+        netuid=netuid,
+        amount=Balance.from_tao(1_000),
+        wait_for_inclusion=True,
+        wait_for_finalization=True,
+    ), "Failed to add stake for Alice"
+
+    # Wait for the first epoch to pass
+    await wait_epoch(subtensor, netuid)
+
+    # Add further stake so validator permit is activated
+    assert subtensor.add_stake(
+        alice_wallet,
+        netuid=netuid,
+        amount=Balance.from_tao(1_000),
+        wait_for_inclusion=True,
+        wait_for_finalization=True,
+    ), "Failed to add stake for Alice"
+
     # Get latest metagraph
     metagraph = subtensor.metagraph(netuid)
 
@@ -70,11 +110,11 @@ async def test_incentive(local_chain, subtensor, templates, alice_wallet, bob_wa
 
     async with templates.miner(bob_wallet, netuid):
         async with templates.validator(alice_wallet, netuid):
-            # wait for the Validator to process and set_weights
+            # Wait for the Validator to process and set_weights
             await asyncio.sleep(5)
 
             # Wait few epochs
-            await wait_epoch(subtensor, netuid, times=4)
+            await wait_epoch(subtensor, netuid, times=2)
 
             # Refresh metagraph
             metagraph = subtensor.metagraph(netuid)