metagraph e2e

Author: ibraheem-abe

tests/e2e_tests/test_metagraph.py

@@ -48,7 +48,7 @@ def test_metagraph(local_chain):
         AssertionError: If any of the checks or verifications fail
     """
     logging.console.info("Testing test_metagraph_command")
-    netuid = 1
+    netuid = 2
 
     # Register Alice, Bob, and Dave
     alice_keypair, alice_wallet = setup_wallet("//Alice")
@@ -60,15 +60,15 @@ def test_metagraph(local_chain):
 
     # Verify subnet was created successfully
     assert local_chain.query(
-        "SubtensorModule", "NetworksAdded", [1]
+        "SubtensorModule", "NetworksAdded", [2]
     ).serialize(), "Subnet wasn't created successfully"
 
     # Initialize metagraph
     subtensor = Subtensor(network="ws://localhost:9945")
-    metagraph = subtensor.metagraph(netuid=1)
+    metagraph = subtensor.metagraph(netuid=2)
 
     # Assert metagraph is empty
-    assert len(metagraph.uids) == 0, "Metagraph is not empty"
+    assert len(metagraph.uids) == 1, "Metagraph is not empty"
 
     # Register Bob to the subnet
     assert subtensor.burned_register(
@@ -79,14 +79,14 @@ def test_metagraph(local_chain):
     metagraph.sync(subtensor=subtensor)
 
     # Assert metagraph has Bob neuron
-    assert len(metagraph.uids) == 1, "Metagraph doesn't have exactly 1 neuron"
+    assert len(metagraph.uids) == 2, "Metagraph doesn't have exactly 2 neurons"
     assert (
-        metagraph.hotkeys[0] == bob_keypair.ss58_address
+        metagraph.hotkeys[1] == bob_keypair.ss58_address
     ), "Bob's hotkey doesn't match in metagraph"
-    assert len(metagraph.coldkeys) == 1, "Metagraph doesn't have exactly 1 coldkey"
-    assert metagraph.n.max() == 1, "Metagraph's max n is not 1"
-    assert metagraph.n.min() == 1, "Metagraph's min n is not 1"
-    assert len(metagraph.addresses) == 1, "Metagraph doesn't have exactly 1 address"
+    assert len(metagraph.coldkeys) == 2, "Metagraph doesn't have exactly 2 coldkeys"
+    assert metagraph.n.max() == 2, "Metagraph's max n is not 2"
+    assert metagraph.n.min() == 2, "Metagraph's min n is not 2"
+    assert len(metagraph.addresses) == 2, "Metagraph doesn't have exactly 2 addresses"
 
     # Fetch UID of Bob
     uid = subtensor.get_uid_for_hotkey_on_subnet(
@@ -104,7 +104,7 @@ def test_metagraph(local_chain):
     ), "Neuron info of Bob doesn't match b/w metagraph & subtensor"
 
     # Create pre_dave metagraph for future verifications
-    metagraph_pre_dave = subtensor.metagraph(netuid=1)
+    metagraph_pre_dave = subtensor.metagraph(netuid=2)
 
     # Register Dave as a neuron
     assert subtensor.burned_register(
@@ -115,32 +115,32 @@ def test_metagraph(local_chain):
 
     # Assert metagraph now includes Dave's neuron
     assert (
-        len(metagraph.uids) == 2
-    ), "Metagraph doesn't have exactly 2 neurons post Dave"
+        len(metagraph.uids) == 3
+    ), "Metagraph doesn't have exactly 3 neurons post Dave"
     assert (
-        metagraph.hotkeys[1] == dave_keypair.ss58_address
+        metagraph.hotkeys[2] == dave_keypair.ss58_address
     ), "Neuron's hotkey in metagraph doesn't match"
     assert (
-        len(metagraph.coldkeys) == 2
-    ), "Metagraph doesn't have exactly 2 coldkeys post Dave"
-    assert metagraph.n.max() == 2, "Metagraph's max n is not 2 post Dave"
-    assert metagraph.n.min() == 2, "Metagraph's min n is not 2 post Dave"
-    assert len(metagraph.addresses) == 2, "Metagraph doesn't have 2 addresses post Dave"
+        len(metagraph.coldkeys) == 3
+    ), "Metagraph doesn't have exactly 3 coldkeys post Dave"
+    assert metagraph.n.max() == 3, "Metagraph's max n is not 3 post Dave"
+    assert metagraph.n.min() == 3, "Metagraph's min n is not 3 post Dave"
+    assert len(metagraph.addresses) == 3, "Metagraph doesn't have 3 addresses post Dave"
 
     # Test staking with low balance
     assert not add_stake(
-        local_chain, dave_wallet, Balance.from_tao(10_000)
+        local_chain, dave_wallet, Balance.from_tao(10_000), netuid=netuid
     ), "Low balance stake should fail"
 
     # Add stake by Bob
     assert add_stake(
-        local_chain, bob_wallet, Balance.from_tao(10_000)
+        local_chain, bob_wallet, Balance.from_tao(10_000), netuid=netuid
     ), "Failed to add stake for Bob"
 
     # Assert stake is added after updating metagraph
     metagraph.sync(subtensor=subtensor)
-    assert metagraph.neurons[0].stake == Balance.from_tao(
-        10_000
+    assert metagraph.neurons[1].stake > Balance.from_tao(
+        0
     ), "Bob's stake not updated in metagraph"
 
     # Test the save() and load() mechanism

tests/e2e_tests/utils/chain_interactions.py

@@ -71,15 +71,15 @@ def sudo_set_hyperparameter_values(
 
 
 def add_stake(
-    substrate: "SubstrateInterface", wallet: "Wallet", amount: "Balance"
+    substrate: "SubstrateInterface", wallet: "Wallet", amount: "Balance", netuid: int
 ) -> bool:
     """
     Adds stake to a hotkey using SubtensorModule. Mimics command of adding stake
     """
     stake_call = substrate.compose_call(
         call_module="SubtensorModule",
         call_function="add_stake",
-        call_params={"hotkey": wallet.hotkey.ss58_address, "amount_staked": amount.rao},
+        call_params={"hotkey": wallet.hotkey.ss58_address, "amount_staked": amount.rao, "netuid": netuid},
     )
     extrinsic = substrate.create_signed_extrinsic(
         call=stake_call, keypair=wallet.coldkey