Test Staking

Author: thewhaleking

bittensor_cli/cli.py

@@ -4501,7 +4501,7 @@ def sudo_get(
 
         [green]$[/green] btcli sudo get --netuid 1
         """
-        self.verbosity_handler(quiet, verbose)
+        self.verbosity_handler(quiet, verbose, json_output)
         return self._run_command(
             sudo.get_hyperparameters(
                 self.initialize_chain(network), netuid, json_output

bittensor_cli/src/commands/stake/list.py

@@ -229,8 +229,8 @@ def create_table(hotkey_: str, substakes: list[StakeInfo]):
                     {
                         "netuid": netuid,
                         "subnet_name": subnet_name,
-                        "value": tao_value.tao,
-                        "stake_value": stake_value,
+                        "value": tao_value_.tao,
+                        "stake_value": substake_.stake.tao,
                         "rate": pool.price.tao,
                         "swap_value": swap_value,
                         "registered": True if substake_.is_registered else False,

bittensor_cli/src/commands/sudo.py

@@ -657,7 +657,8 @@ async def get_hyperparameters(
         )
     if json_output:
         json_console.print(json.dumps(dict_out))
-    console.print(table)
+    else:
+        console.print(table)
     return True
 
 

tests/e2e_tests/test_staking_sudo.py

@@ -1,3 +1,4 @@
+import json
 import re
 
 from bittensor_cli.src.bittensor.balances import Balance
@@ -129,8 +130,10 @@ def test_staking(local_chain, wallet_setup):
             wallet_alice.name,
             "--chain",
             "ws://127.0.0.1:9945",
+            "--verbose",
         ],
     )
+
     # Assert correct stake is added
     cleaned_stake = [
         re.sub(r"\s+", " ", line) for line in show_stake.stdout.splitlines()
@@ -151,6 +154,9 @@ def test_staking(local_chain, wallet_setup):
             "--json-output",
         ],
     )
+    show_stake_json_output = json.loads(show_stake_json.stdout)
+    alice_stake = show_stake_json_output["stake_info"][keypair_alice.ss58_address][0]
+    assert Balance.from_tao(alice_stake["stake_value"]) > Balance.from_tao(90.0)
 
     # Execute remove_stake command and remove all alpha stakes from Alice
     remove_stake = exec_command_alice(
@@ -196,7 +202,24 @@ def test_staking(local_chain, wallet_setup):
     max_burn_tao = all_hyperparams[22].split()[3]
 
     # Assert max_burn is 100 TAO from default
-    assert Balance.from_tao(float(max_burn_tao)) == Balance.from_tao(100)
+    assert Balance.from_tao(float(max_burn_tao)) == Balance.from_tao(100.0)
+
+    hyperparams_json = exec_command_alice(
+        command="sudo",
+        sub_command="get",
+        extra_args=[
+            "--chain",
+            "ws://127.0.0.1:9945",
+            "--netuid",
+            netuid,
+            "--json-output",
+        ],
+    )
+    hyperparams_json_output = json.loads(hyperparams_json.stdout)
+    max_burn_tao_from_json = next(
+        filter(lambda x: x["hyperparameter"] == "max_burn", hyperparams_json_output)
+    )["value"]
+    assert Balance.from_rao(max_burn_tao_from_json) == Balance.from_tao(100.0)
 
     # Change max_burn hyperparameter to 10 TAO
     change_hyperparams = exec_command_alice(
@@ -241,4 +264,24 @@ def test_staking(local_chain, wallet_setup):
 
     # Assert max_burn is now 10 TAO
     assert Balance.from_tao(float(updated_max_burn_tao)) == Balance.from_tao(10)
+
+    updated_hyperparams_json = exec_command_alice(
+        command="sudo",
+        sub_command="get",
+        extra_args=[
+            "--chain",
+            "ws://127.0.0.1:9945",
+            "--netuid",
+            netuid,
+            "--json-output",
+        ],
+    )
+    updated_hyperparams_json_output = json.loads(updated_hyperparams_json.stdout)
+    max_burn_tao_from_json = next(
+        filter(
+            lambda x: x["hyperparameter"] == "max_burn", updated_hyperparams_json_output
+        )
+    )["value"]
+    assert Balance.from_rao(max_burn_tao_from_json) == Balance.from_tao(10.0)
+
     print("✅ Passed staking and sudo commands")