Merge pull request #404 from opentensor/fix/e2e-tests/260325

Fixes staking/unstaking e2e tests

Author: ibraheem-abe

tests/e2e_tests/test_staking_sudo.py

@@ -136,7 +136,7 @@ def test_staking(local_chain, wallet_setup):
     stake_added = cleaned_stake[8].split("│")[3].strip().split()[0]
     assert Balance.from_tao(float(stake_added)) >= Balance.from_tao(90)
 
-    # Execute remove_stake command and remove all 100 TAO from Alice
+    # Execute remove_stake command and remove all alpha stakes from Alice
     remove_stake = exec_command_alice(
         command="stake",
         sub_command="remove",
@@ -152,7 +152,7 @@ def test_staking(local_chain, wallet_setup):
             "--chain",
             "ws://127.0.0.1:9945",
             "--amount",
-            "100",
+            str(float(stake_added) - 1),
             "--tolerance",
             "0.1",
             "--partial",

tests/e2e_tests/test_unstaking.py

@@ -208,21 +208,11 @@ def test_unstaking(local_chain, wallet_setup):
     cleaned_stake = [
         re.sub(r"\s+", " ", line) for line in stake_list.stdout.splitlines()
     ]
-    stake_after_unstaking_netuid_2 = cleaned_stake[9].split("│")[3].strip().split()[0]
+    stake_after_unstaking_netuid_2 = cleaned_stake[10].split("│")[3].strip().split()[0]
     assert Balance.from_tao(float(stake_after_unstaking_netuid_2)) <= Balance.from_tao(
         float(inital_stake_netuid_2)
     )
 
-    print(
-        "Bob wallet deets: ",
-        "hotkey: ",
-        wallet_bob.hotkey_str,
-        "name: ",
-        wallet_bob.name,
-        "path: ",
-        wallet_bob.path,
-    )
-
     # Remove all alpha stakes
     unstake_alpha = exec_command_bob(
         command="stake",