fix staking e2e test with new changes in the chain

Author: Roman

tests/e2e_tests/test_staking.py

@@ -223,13 +223,17 @@ def test_batch_operations(subtensor, alice_wallet, bob_wallet):
         bob_wallet.coldkey.ss58_address,
     )
 
-    assert balances == {
+    expected_balances = {
         alice_wallet.coldkey.ss58_address: get_dynamic_balance(
-            balances[alice_wallet.coldkey.ss58_address].rao, 2
+            balances[alice_wallet.coldkey.ss58_address].rao
+        ),
+        bob_wallet.coldkey.ss58_address: get_dynamic_balance(
+            balances[bob_wallet.coldkey.ss58_address].rao
         ),
-        bob_wallet.coldkey.ss58_address: Balance.from_tao(999_998).set_unit(3),
     }
 
+    assert balances == expected_balances
+
     alice_balance = balances[alice_wallet.coldkey.ss58_address]
 
     success = subtensor.add_stake_multiple(
@@ -260,11 +264,17 @@ def test_batch_operations(subtensor, alice_wallet, bob_wallet):
         bob_wallet.coldkey.ss58_address,
     )
 
-    assert balances == {
-        alice_wallet.coldkey.ss58_address: ApproxBalance(alice_balance.rao),
-        bob_wallet.coldkey.ss58_address: Balance.from_tao(999_998),
+    expected_balances = {
+        alice_wallet.coldkey.ss58_address: get_dynamic_balance(
+            balances[alice_wallet.coldkey.ss58_address].rao
+        ),
+        bob_wallet.coldkey.ss58_address: get_dynamic_balance(
+            balances[bob_wallet.coldkey.ss58_address].rao
+        ),
     }
 
+    assert balances == expected_balances
+
     success = subtensor.unstake_multiple(
         alice_wallet,
         hotkey_ss58s=[bob_wallet.hotkey.ss58_address for _ in netuids],
@@ -288,12 +298,15 @@ def test_batch_operations(subtensor, alice_wallet, bob_wallet):
         bob_wallet.coldkey.ss58_address,
     )
 
-    assert balances == {
+    expected_balances = {
         alice_wallet.coldkey.ss58_address: get_dynamic_balance(
-            balances[alice_wallet.coldkey.ss58_address].rao, 2
+            balances[alice_wallet.coldkey.ss58_address].rao,
         ),
-        bob_wallet.coldkey.ss58_address: Balance.from_tao(999_998),
+        bob_wallet.coldkey.ss58_address: Balance.from_tao(999_999.8),
     }
+
+    assert balances == expected_balances
+
     assert balances[alice_wallet.coldkey.ss58_address] > alice_balance
     logging.console.success(f"✅ Test [green]test_batch_operations[/green] passed")