fix for unit tests

Author: basfroman

tests/unit_tests/extrinsics/test_staking.py

@@ -181,6 +181,7 @@ def test_add_stake_multiple_extrinsic(subtensor, mocker, fake_wallet):
         "get_stake_info_for_coldkey",
         return_value=[Balance(1.1), Balance(0.3)],
     )
+    mocker.patch.object(subtensor, "get_stake", return_value=Balance.from_tao(2.2))
     mocker.patch.object(subtensor, "get_balance", return_value=Balance.from_tao(10))
     mocker.patch.object(
         staking, "get_old_stakes", return_value=[Balance(1.1), Balance(0.3)]

tests/unit_tests/extrinsics/test_unstaking.py

@@ -119,7 +119,8 @@ def test_unstake_multiple_extrinsic(subtensor, fake_wallet, mocker):
     mocked_balance = mocker.patch.object(
         subtensor, "get_balance", return_value=Balance.from_tao(1.0)
     )
-    mocked_get_stake_for_coldkey_and_hotkey = mocker.patch.object(
+    mocker.patch.object(subtensor, "get_stake", return_value=Balance.from_tao(8.9))
+    mocker.patch.object(
         subtensor, "get_stake_for_coldkey_and_hotkey", return_value=[Balance(10.0)]
     )
     mocked_unstake_extrinsic = mocker.patch.object(