tests fix

Author: basfroman

bittensor/core/subtensor.py

@@ -16,6 +16,7 @@
 from scalecodec.exceptions import RemainingScaleBytesNotEmptyException
 from scalecodec.type_registry import load_type_registry_preset
 from scalecodec.types import ScaleType
+from substrateinterface import Keypair
 from substrateinterface.base import QueryMapResult, SubstrateInterface
 from websockets.exceptions import InvalidStatus
 from websockets.sync import client as ws_client
@@ -1532,9 +1533,11 @@ def get_transfer_fee(
                 call_params={"dest": dest, "value": value.rao},
             )
 
+            temp_keypair = Keypair(ss58_address=wallet.coldkeypub.ss58_address)
+
             try:
                 payment_info = self.substrate.get_payment_info(
-                    call=call, keypair=wallet.coldkeypub
+                    call=call, keypair=temp_keypair
                 )
             except Exception as e:
                 logging.error(f"[red]Failed to get payment info.[/red] {e}")

tests/e2e_tests/test_transfer.py

@@ -1,6 +1,9 @@
 from bittensor.core.subtensor import Subtensor
 from bittensor.utils.balance import Balance
 from tests.e2e_tests.utils.e2e_test_utils import setup_wallet
+from bittensor import logging
+
+logging.set_trace()
 
 
 def test_transfer(local_chain):

tests/unit_tests/test_subtensor.py

@@ -1739,6 +1739,8 @@ def test_get_transfer_fee(subtensor, mocker):
     fake_payment_info = {"partialFee": int(2e10)}
     subtensor.substrate.get_payment_info.return_value = fake_payment_info
 
+    mocker.patch.object(subtensor_module, "Keypair", return_value=mocker.MagicMock())
+
     # Call
     result = subtensor.get_transfer_fee(wallet=fake_wallet, dest=fake_dest, value=value)
 
@@ -1751,7 +1753,7 @@ def test_get_transfer_fee(subtensor, mocker):
 
     subtensor.substrate.get_payment_info.assert_called_once_with(
         call=subtensor.substrate.compose_call.return_value,
-        keypair=fake_wallet.coldkeypub,
+        keypair=subtensor_module.Keypair.return_value,
     )
 
     assert result == 2e10