Updates tests

Author: ibraheem-abe

tests/e2e_tests/test_staking_sudo.py

@@ -298,7 +298,7 @@ def test_staking(local_chain, wallet_setup):
 
     # Parse all hyperparameters and single out max_burn in TAO
     all_hyperparams = hyperparams.stdout.splitlines()
-    max_burn_tao = all_hyperparams[22].split()[3]
+    max_burn_tao = all_hyperparams[22].split()[2].strip('\u200e')
 
     # Assert max_burn is 100 TAO from default
     assert Balance.from_tao(float(max_burn_tao)) == Balance.from_tao(100.0)
@@ -359,7 +359,7 @@ 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()[3]
+    updated_max_burn_tao = all_updated_hyperparams[22].split()[2].strip('\u200e')
 
     # Assert max_burn is now 10 TAO
     assert Balance.from_tao(float(updated_max_burn_tao)) == Balance.from_tao(10)

tests/e2e_tests/utils.py

@@ -57,7 +57,7 @@ def exec_command(
     return keypair, wallet, wallet_path, exec_command
 
 
-def extract_coldkey_balance(text: str, wallet_name: str, coldkey_address: str) -> dict:
+def extract_coldkey_balance(cleaned_text: str, wallet_name: str, coldkey_address: str) -> dict:
     """
     Extracts the free, staked, and total balances for a
     given wallet name and coldkey address from the input string.
@@ -72,11 +72,12 @@ def extract_coldkey_balance(text: str, wallet_name: str, coldkey_address: str) -
               each containing the corresponding balance as a Balance object.
               Returns a dictionary with all zeros if the wallet name or coldkey address is not found.
     """
+    cleaned_text = cleaned_text.replace('\u200e', '')
     pattern = (
-        rf"{wallet_name}\s+{coldkey_address}\s+" r"τ\s*([\d,]+\.\d+)"  # Free Balance
+        rf"{wallet_name}\s+{coldkey_address}\s+([\d,]+\.\d+)\s*τ"  # Free Balance
     )
 
-    match = re.search(pattern, text)
+    match = re.search(pattern, cleaned_text)
 
     if not match:
         return {