test: set takes

tests/e2e_tests/subcommands/delegation/test_set_delegate_take.py

@@ -0,0 +1,56 @@
+from bittensor.commands.delegates import SetTakeCommand, NominateCommand
+from bittensor.commands.network import RegisterSubnetworkCommand
+from bittensor.commands.register import RegisterCommand
+from bittensor.commands.root import RootRegisterCommand
+
+from tests.e2e_tests.utils import setup_wallet
+
+
+def test_set_delegate_increase_take(local_chain):
+    # Register root as Alice
+    (keypair, exec_command) = setup_wallet("//Alice")
+    exec_command(RootRegisterCommand, ["root", "register"])
+
+    # Create subnet 1 and verify created successfully
+    assert not (local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize())
+
+    exec_command(RegisterSubnetworkCommand, ["s", "create"])
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [1])
+
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [1]).serialize()
+
+    # Register and nominate Bob
+    (keypair, exec_command) = setup_wallet("//Bob")
+    assert (
+        local_chain.query(
+            "SubtensorModule", "LastTxBlock", [keypair.ss58_address]
+        ).serialize()
+        == 0
+    )
+
+    assert (
+        local_chain.query(
+            "SubtensorModule", "LastTxBlockDelegateTake", [keypair.ss58_address]
+        ).serialize()
+        == 0
+    )
+    exec_command(RegisterCommand, ["s", "register", "--netuid", "1"])
+    exec_command(NominateCommand, ["root", "nominate"])
+    assert (
+        local_chain.query(
+            "SubtensorModule", "LastTxBlock", [keypair.ss58_address]
+        ).serialize()
+        > 0
+    )
+    assert (
+        local_chain.query(
+            "SubtensorModule", "LastTxBlockDelegateTake", [keypair.ss58_address]
+        ).serialize()
+        > 0
+    )
+
+    # Set delegate take for Bob
+    exec_command(SetTakeCommand, ["r", "set_take", "--take", "0.15"])
+    assert local_chain.query(
+        "SubtensorModule", "Delegates", [keypair.ss58_address]
+    ).value == int(0.15 * 65535)