Tests/e2e tests staging

bittensor/subtensor.py

@@ -4270,9 +4270,9 @@ def make_substrate_call_with_retry(encoded_hotkey_: List[int]):
 
             return self.substrate.rpc_request(
                 method="delegateInfo_getDelegate",  # custom rpc method
-                params=[encoded_hotkey_, block_hash]
-                if block_hash
-                else [encoded_hotkey_],
+                params=(
+                    [encoded_hotkey_, block_hash] if block_hash else [encoded_hotkey_]
+                ),
             )
 
         encoded_hotkey = ss58_to_vec_u8(hotkey_ss58)
@@ -4376,9 +4376,9 @@ def make_substrate_call_with_retry(encoded_coldkey_: List[int]):
 
             return self.substrate.rpc_request(
                 method="delegateInfo_getDelegated",
-                params=[block_hash, encoded_coldkey_]
-                if block_hash
-                else [encoded_coldkey_],
+                params=(
+                    [block_hash, encoded_coldkey_] if block_hash else [encoded_coldkey_]
+                ),
             )
 
         encoded_coldkey = ss58_to_vec_u8(coldkey_ss58)

tests/e2e_tests/multistep/test_axon.py

@@ -19,7 +19,7 @@
 @pytest.mark.asyncio
 async def test_axon(local_chain):
     # Register root as Alice
-    alice_keypair, exec_command, wallet_path = setup_wallet("//Alice")
+    alice_keypair, exec_command, wallet_path = setup_wallet("//Alice", True)
     exec_command(RegisterSubnetworkCommand, ["s", "create"])
 
     # Verify subnet 1 created successfully

tests/e2e_tests/multistep/test_dendrite.py

@@ -25,13 +25,13 @@
 @pytest.mark.asyncio
 async def test_dendrite(local_chain):
     # Register root as Alice - the subnet owner
-    alice_keypair, exec_command, wallet_path = setup_wallet("//Alice")
+    alice_keypair, exec_command, wallet_path = setup_wallet("//Alice", True)
     exec_command(RegisterSubnetworkCommand, ["s", "create"])
 
     # Verify subnet 1 created successfully
     assert local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
 
-    bob_keypair, exec_command, wallet_path = setup_wallet("//Bob")
+    bob_keypair, exec_command, wallet_path = setup_wallet("//Bob", True)
 
     # Register a neuron to the subnet
     exec_command(

tests/e2e_tests/multistep/test_last_tx_block.py

@@ -9,7 +9,7 @@
 # https://discord.com/channels/799672011265015819/1176889736636407808/1236057424134144152
 def test_takes(local_chain):
     # Register root as Alice
-    keypair, exec_command, wallet_path = setup_wallet("//Alice")
+    keypair, exec_command, wallet_path = setup_wallet("//Alice", True)
     exec_command(RootRegisterCommand, ["root", "register"])
 
     # Create subnet 1 and verify created successfully
@@ -21,7 +21,7 @@ def test_takes(local_chain):
     assert local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
 
     # Register and nominate Bob
-    keypair, exec_command, wallet_path = setup_wallet("//Bob")
+    keypair, exec_command, wallet_path = setup_wallet("//Bob", True)
     assert (
         local_chain.query(
             "SubtensorModule", "LastTxBlock", [keypair.ss58_address]

tests/e2e_tests/subcommands/delegation/test_set_delegate_take.py

@@ -8,7 +8,7 @@
 
 def test_set_delegate_increase_take(local_chain):
     # Register root as Alice
-    keypair, exec_command, wallet_path = setup_wallet("//Alice")
+    keypair, exec_command, wallet_path = setup_wallet("//Alice", True)
     exec_command(RootRegisterCommand, ["root", "register"])
 
     # Create subnet 1 and verify created successfully
@@ -20,7 +20,7 @@ def test_set_delegate_increase_take(local_chain):
     assert local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
 
     # Register and nominate Bob
-    keypair, exec_command, wallet_path = setup_wallet("//Bob")
+    keypair, exec_command, wallet_path = setup_wallet("//Bob", True)
     assert (
         local_chain.query(
             "SubtensorModule", "LastTxBlock", [keypair.ss58_address]

tests/e2e_tests/subcommands/root/test_root_boost_weights.py

@@ -0,0 +1,59 @@
+from bittensor.commands.root import RootSetBoostCommand, RootSetWeightsCommand
+from bittensor.commands.stake import StakeCommand
+from bittensor.commands.unstake import UnStakeCommand
+from bittensor.commands.network import RegisterSubnetworkCommand
+from bittensor.commands.register import RegisterCommand
+from ...utils import new_wallet, sudo_call_set_network_limit
+import bittensor
+
+
+# we can't test it now since the root network weights can't be set
+# Test case to set weights for root network
+def test_root_get_set_weights(local_chain, capsys):
+    (wallet, exec_command) = new_wallet("//Alice", "//Bob")
+    assert sudo_call_set_network_limit(local_chain, wallet)
+
+    assert not local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
+
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
+
+    # assert (
+    #     local_chain.query("SubtensorModule", "Uids", [1, wallet.hotkey.ss58_address])
+    #     == None
+    # )
+
+    # exec_command(RegisterCommand, ["subnets", "register", "--netuid", "0"])
+
+    # can not set weights for root network. update needed from python implementation
+    # netuids = "1,2,4"
+    # weights = "0.1,0.3,0.6"
+    # exec_command(
+    #     RootSetWeightsCommand,
+    #     ["root", "weights", "--netuids", netuids, "--weights", weights],
+    # )
+
+    # weights = local_chain.query_map(
+    #     "SubtensorModule", "Weights", [wallet.hotkey.ss58_address]
+    # )
+
+    # netuid = "1"
+    # increase = "0.01"
+
+    # exec_command(
+    #     RootSetBoostCommand,
+    #     ["root", "boost", "--netuid", netuid, "--increase", increase],
+    # )
+
+    # weights = local_chain.query("SubtensorModule", "Weights", [1])
+    # assert weights == 1
+
+    # captured = capsys.readouterr()
+    # lines = captured.out.splitlines()
+
+    # for line in lines:
+    #     bittensor.logging.info(line)
+
+    # assert len(lines) == 4

tests/e2e_tests/subcommands/root/test_root_delegate_list.py

@@ -0,0 +1,21 @@
+from bittensor.commands.delegates import ListDelegatesCommand
+from bittensor.commands.root import RootRegisterCommand
+from bittensor.commands.delegates import SetTakeCommand
+from ...utils import new_wallet
+
+
+# delegate seems hard code the network config
+def test_root_delegate_list(local_chain, capsys):
+    (wallet, exec_command) = new_wallet("//Alice", "//Bob")
+
+    # 1200 hardcoded block gap
+    exec_command(
+        ListDelegatesCommand,
+        ["root", "list_delegates"],
+    )
+
+    captured = capsys.readouterr()
+    lines = captured.out.splitlines()
+
+    # the command print too many lines
+    assert len(lines) > 200

tests/e2e_tests/subcommands/root/test_root_delegate_stake_unstake.py

@@ -0,0 +1,101 @@
+from bittensor.commands.delegates import DelegateStakeCommand, DelegateUnstakeCommand
+from bittensor.commands.stake import StakeCommand
+from bittensor.commands.delegates import SetTakeCommand
+from bittensor.commands.network import RegisterSubnetworkCommand
+from bittensor.commands.register import RegisterCommand
+
+from ...utils import (
+    new_wallet,
+    call_add_proposal,
+    sudo_call_set_network_limit,
+    sudo_call_set_target_stakes_per_interval,
+)
+
+
+# delegate seems hard code the network config
+def test_root_delegate_stake(local_chain, capsys):
+    (wallet, exec_command) = new_wallet("//Alice", "//Bob")
+
+    stakes = local_chain.query(
+        "SubtensorModule",
+        "Stake",
+        [wallet.hotkey.ss58_address, wallet.coldkey.ss58_address],
+    )
+    assert stakes == 0
+
+    assert sudo_call_set_network_limit(local_chain, wallet)
+    assert sudo_call_set_target_stakes_per_interval(local_chain, wallet)
+
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    exec_command(RegisterCommand, ["s", "register", "--neduid", "1"])
+
+    stake_amount = 2
+    exec_command(StakeCommand, ["stake", "add", "--amount", str(stake_amount)])
+
+    stakes = local_chain.query(
+        "SubtensorModule",
+        "Stake",
+        [wallet.hotkey.ss58_address, wallet.coldkey.ss58_address],
+    )
+
+    assert stakes > 1_000_000_000
+
+    delegates = local_chain.query(
+        "SubtensorModule",
+        "Delegates",
+        [wallet.hotkey.ss58_address],
+    )
+
+    assert delegates == 11796
+
+    # stake 1 TAO
+    # exec_command(
+    #     DelegateStakeCommand,
+    #     [
+    #         "root",
+    #         "delegate",
+    #         "--delegate_ss58key",
+    #         wallet.hotkey.ss58_address,
+    #         "--amount",
+    #         "1",
+    #     ],
+    # )
+
+    # new_stakes = local_chain.query(
+    #     "SubtensorModule",
+    #     "Stake",
+    #     [wallet.hotkey.ss58_address, wallet.coldkey.ss58_address],
+    # )
+
+    # tolerance = 10000
+
+    # assert (
+    #     stakes.serialize() + 1_000_000_000 - tolerance
+    #     < new_stakes.serialize()
+    #     < stakes.serialize() + 1_000_000_000 + tolerance
+    # )
+
+    # unstake 1 TAO
+    # exec_command(
+    #     DelegateUnstakeCommand,
+    #     [
+    #         "root",
+    #         "delegate",
+    #         "--delegate_ss58key",
+    #         wallet.hotkey.ss58_address,
+    #         "--amount",
+    #         "1",
+    #     ],
+    # )
+
+    # stakes = local_chain.query(
+    #     "SubtensorModule",
+    #     "Stake",
+    #     [wallet.hotkey.ss58_address, wallet.coldkey.ss58_address],
+    # )
+
+    # assert (
+    #     stakes.serialize() + 1_000_000_000 - tolerance
+    #     < new_stakes.serialize()
+    #     < stakes.serialize() + 1_000_000_000 + tolerance
+    # )

tests/e2e_tests/subcommands/root/test_root_get_set_weights.py

@@ -0,0 +1,38 @@
+from bittensor.commands.root import RootSetWeightsCommand, RootGetWeightsCommand
+from bittensor.commands.network import RegisterSubnetworkCommand
+from ...utils import new_wallet, sudo_call_set_network_limit
+
+
+# we can't test it now since the root network weights can't be set
+# Test case to set weights for root network and get the weights
+def test_root_get_set_weights(local_chain, capsys):
+    (wallet, exec_command) = new_wallet("//Alice", "//Bob")
+    assert sudo_call_set_network_limit(local_chain, wallet)
+    assert not local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
+
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [2]).serialize()
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [4]).serialize()
+
+    netuids = "1,2,4"
+    weights = "0.1,0.3,0.6"
+
+    # this command need update, should set the netuid. subtensor not accept the weight set for root network
+    exec_command(
+        RootSetWeightsCommand,
+        ["root", "weights", "--netuids", netuids, "--weights", weights],
+    )
+
+    weights = local_chain.query("SubtensorModule", "Weights", [1, 0])
+
+    exec_command(
+        RootGetWeightsCommand,
+        ["root", "get_weights"],
+    )
+
+    captured = capsys.readouterr()
+    lines = captured.out.splitlines()

tests/e2e_tests/subcommands/root/test_root_list.py

@@ -0,0 +1,30 @@
+from bittensor.commands.root import RootList
+from ...utils import new_wallet
+from bittensor.commands.network import RegisterSubnetworkCommand
+import bittensor
+
+
+# test case to list the root network
+def test_root_list(local_chain, capsys):
+    (wallet, exec_command) = new_wallet("//Alice", "//Bob")
+
+    exec_command(RootList, ["root", "list"])
+    captured = capsys.readouterr()
+    lines = captured.out.split("\n")
+
+    assert len(lines) >= 4
+    bittensor.logging.info(lines)
+
+    # assert "Root Network" in lines[0]
+    # assert "UID  NAME  ADDRESS  STAKE" in lines[1]
+
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+
+    exec_command(RootList, ["root", "list"])
+    captured = capsys.readouterr()
+    lines = captured.out.splitlines()
+
+    assert len(lines) >= 4
+    # assert "Root Network" in lines[0]
+    # assert "UID  NAME  ADDRESS  STAKE" in lines[1]
+    # assert "1" in lines[2]

tests/e2e_tests/subcommands/root/test_root_nominate.py

@@ -0,0 +1,47 @@
+from bittensor.commands.delegates import NominateCommand
+from bittensor.commands.stake import StakeCommand
+from bittensor.commands.delegates import SetTakeCommand
+from bittensor.commands.network import RegisterSubnetworkCommand
+from bittensor.commands.register import RegisterCommand
+
+from ...utils import (
+    new_wallet,
+    call_add_proposal,
+    sudo_call_set_network_limit,
+    sudo_call_set_target_stakes_per_interval,
+)
+
+
+# delegate seems hard code the network config
+def test_root_nominate(local_chain, capsys):
+    (wallet, exec_command) = new_wallet("//Alice", "//Bob")
+
+    delegates = local_chain.query(
+        "SubtensorModule",
+        "Delegates",
+        [wallet.hotkey.ss58_address],
+    )
+
+    assert delegates == 11796
+
+    assert sudo_call_set_network_limit(local_chain, wallet)
+    assert sudo_call_set_target_stakes_per_interval(local_chain, wallet)
+
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    exec_command(RegisterCommand, ["s", "register", "--neduid", "1"])
+
+    exec_command(
+        NominateCommand,
+        [
+            "root",
+            "nominate",
+        ],
+    )
+
+    delegates = local_chain.query(
+        "SubtensorModule",
+        "Delegates",
+        [wallet.hotkey.ss58_address],
+    )
+
+    assert delegates == 11796