Fix tests

Author: thewhaleking

tests/e2e_tests/test_staking_sudo.py

@@ -396,20 +396,17 @@ def test_staking(local_chain, wallet_setup):
     hyperparams = exec_command_alice(
         command="sudo",
         sub_command="get",
-        extra_args=[
-            "--chain",
-            "ws://127.0.0.1:9945",
-            "--netuid",
-            netuid,
-        ],
+        extra_args=["--chain", "ws://127.0.0.1:9945", "--netuid", netuid, "--json-out"],
     )
 
     # Parse all hyperparameters and single out max_burn in TAO
-    all_hyperparams = hyperparams.stdout.splitlines()
-    max_burn_tao = all_hyperparams[22].split()[2].strip("\u200e")
+    all_hyperparams = json.loads(hyperparams.stdout)
+    max_burn_tao = next(
+        filter(lambda x: x["hyperparameter"] == "max_burn", all_hyperparams)
+    )["value"]
 
     # Assert max_burn is 100 TAO from default
-    assert Balance.from_tao(float(max_burn_tao)) == Balance.from_tao(100.0)
+    assert Balance.from_rao(int(max_burn_tao)) == Balance.from_tao(100.0)
 
     hyperparams_json = exec_command_alice(
         command="sudo",
@@ -468,7 +465,11 @@ def test_staking(local_chain, wallet_setup):
 
     # Parse updated hyperparameters
     all_updated_hyperparams = updated_hyperparams.stdout.splitlines()
-    updated_max_burn_tao = all_updated_hyperparams[22].split()[2].strip("\u200e")
+    updated_max_burn_tao = (
+        next(filter(lambda x: x[3:11] == "max_burn", all_updated_hyperparams))
+        .split()[2]
+        .strip("\u200e")
+    )
 
     # Assert max_burn is now 10 TAO
     assert Balance.from_tao(float(updated_max_burn_tao)) == Balance.from_tao(10)