move async extrinsics tests to proper directory

Author: Roman

tests/unit_tests/extrinsics/__init__.py

@@ -1,10 +1,11 @@
 from bittensor.core import async_subtensor as subtensor_module
 from bittensor.core.chain_data import SubnetHyperparameters
 from bittensor.core.async_subtensor import AsyncSubtensor
-from bittensor.core.extrinsics import async_commit_reveal
+from bittensor.core.extrinsics.asyncex import commit_reveal as async_commit_reveal
 import pytest
 import torch
 import numpy as np
+from bittensor_wallet import Wallet
 
 
 @pytest.fixture
@@ -54,7 +55,7 @@ def hyperparams():
 async def test_do_commit_reveal_v3_success(mocker, subtensor):
     """Test successful commit-reveal with wait for finalization."""
     # Preps
-    fake_wallet = mocker.Mock(autospec=subtensor_module.Wallet)
+    fake_wallet = mocker.Mock(autospec=Wallet)
     fake_netuid = 1
     fake_commit = b"fake_commit"
     fake_reveal_round = 1
@@ -90,7 +91,6 @@ async def test_do_commit_reveal_v3_success(mocker, subtensor):
         call=mocked_compose_call.return_value, keypair=fake_wallet.hotkey
     )
     mocked_submit_extrinsic.assert_awaited_once_with(
-        subtensor=subtensor,
         extrinsic=mocked_create_signed_extrinsic.return_value,
         wait_for_inclusion=False,
         wait_for_finalization=False,
@@ -102,7 +102,7 @@ async def test_do_commit_reveal_v3_success(mocker, subtensor):
 async def test_do_commit_reveal_v3_failure_due_to_error(mocker, subtensor):
     """Test commit-reveal fails due to an error in submission."""
     # Preps
-    fake_wallet = mocker.Mock(autospec=subtensor_module.Wallet)
+    fake_wallet = mocker.Mock(autospec=Wallet)
     fake_netuid = 1
     fake_commit = b"fake_commit"
     fake_reveal_round = 1
@@ -149,7 +149,6 @@ async def test_do_commit_reveal_v3_failure_due_to_error(mocker, subtensor):
         call=mocked_compose_call.return_value, keypair=fake_wallet.hotkey
     )
     mocked_submit_extrinsic.assert_awaited_once_with(
-        subtensor=subtensor,
         extrinsic=mocked_create_signed_extrinsic.return_value,
         wait_for_inclusion=True,
         wait_for_finalization=True,
@@ -164,7 +163,7 @@ async def test_commit_reveal_v3_extrinsic_success_with_torch(
 ):
     """Test successful commit-reveal with torch tensors."""
     # Preps
-    fake_wallet = mocker.Mock(autospec=subtensor_module.Wallet)
+    fake_wallet = mocker.Mock(autospec=Wallet)
     fake_netuid = 1
     fake_uids = torch.tensor([1, 2, 3], dtype=torch.int64)
     fake_weights = torch.tensor([0.1, 0.2, 0.7], dtype=torch.float32)
@@ -243,7 +242,7 @@ async def test_commit_reveal_v3_extrinsic_success_with_numpy(
 ):
     """Test successful commit-reveal with numpy arrays."""
     # Preps
-    fake_wallet = mocker.Mock(autospec=subtensor_module.Wallet)
+    fake_wallet = mocker.Mock(autospec=Wallet)
     fake_netuid = 1
     fake_uids = np.array([1, 2, 3], dtype=np.int64)
     fake_weights = np.array([0.1, 0.2, 0.7], dtype=np.float32)
@@ -291,7 +290,7 @@ async def test_commit_reveal_v3_extrinsic_response_false(
 ):
     """Test unsuccessful commit-reveal with torch."""
     # Preps
-    fake_wallet = mocker.Mock(autospec=subtensor_module.Wallet)
+    fake_wallet = mocker.Mock(autospec=Wallet)
     fake_netuid = 1
     fake_uids = torch.tensor([1, 2, 3], dtype=torch.int64)
     fake_weights = torch.tensor([0.1, 0.2, 0.7], dtype=torch.float32)
@@ -348,7 +347,7 @@ async def test_commit_reveal_v3_extrinsic_response_false(
 async def test_commit_reveal_v3_extrinsic_exception(mocker, subtensor):
     """Test exception handling in commit-reveal."""
     # Preps
-    fake_wallet = mocker.Mock(autospec=subtensor_module.Wallet)
+    fake_wallet = mocker.Mock(autospec=Wallet)
     fake_netuid = 1
     fake_uids = [1, 2, 3]
     fake_weights = [0.1, 0.2, 0.7]

tests/unit_tests/extrinsics/asyncex/__init__.py

@@ -2,7 +2,7 @@
 from bittensor_wallet import Wallet
 
 from bittensor.core import async_subtensor
-from bittensor.core.extrinsics import async_registration
+from bittensor.core.extrinsics.asyncex import registration as async_registration
 
 
 @pytest.fixture(autouse=True)

tests/unit_tests/extrinsics/test_async_commit_reveal.py renamed to tests/unit_tests/extrinsics/asyncex/test_commit_reveal.py

@@ -2,7 +2,7 @@
 from substrateinterface.exceptions import SubstrateRequestException
 
 from bittensor.core import async_subtensor
-from bittensor.core.extrinsics import async_root
+from bittensor.core.extrinsics.asyncex import root as async_root
 from bittensor_wallet import Wallet
 
 
@@ -311,7 +311,7 @@ async def test_do_set_root_weights_success(subtensor, mocker):
         era={"period": 5},
     )
     subtensor.substrate.submit_extrinsic.assert_called_once_with(
-        fake_extrinsic, wait_for_inclusion=True, wait_for_finalization=True
+        extrinsic=fake_extrinsic, wait_for_inclusion=True, wait_for_finalization=True
     )
     fake_response.process_events.assert_called_once()
     assert result is True
@@ -405,7 +405,7 @@ async def test_do_set_root_weights_no_waiting(subtensor, mocker):
     subtensor.substrate.compose_call.assert_called_once()
     subtensor.substrate.create_signed_extrinsic.assert_called_once()
     subtensor.substrate.submit_extrinsic.assert_called_once_with(
-        fake_extrinsic, wait_for_inclusion=False, wait_for_finalization=False
+        extrinsic=fake_extrinsic, wait_for_inclusion=False, wait_for_finalization=False
     )
     assert result is True
     assert message == "Not waiting for finalization or inclusion."

tests/unit_tests/extrinsics/test_async_registration.py renamed to tests/unit_tests/extrinsics/asyncex/test_registration.py

@@ -1,7 +1,7 @@
 import pytest
 from bittensor.core import async_subtensor
 from bittensor_wallet import Wallet
-from bittensor.core.extrinsics import async_transfer
+from bittensor.core.extrinsics.asyncex import transfer as async_transfer
 from bittensor.utils.balance import Balance
 
 
@@ -60,7 +60,7 @@ async def test_do_transfer_success(subtensor, mocker):
         call=subtensor.substrate.compose_call.return_value, keypair=fake_wallet.coldkey
     )
     subtensor.substrate.submit_extrinsic.assert_called_once_with(
-        subtensor.substrate.create_signed_extrinsic.return_value,
+        extrinsic=subtensor.substrate.create_signed_extrinsic.return_value,
         wait_for_inclusion=True,
         wait_for_finalization=True,
     )
@@ -119,7 +119,7 @@ async def test_do_transfer_failure(subtensor, mocker):
         call=subtensor.substrate.compose_call.return_value, keypair=fake_wallet.coldkey
     )
     subtensor.substrate.submit_extrinsic.assert_called_once_with(
-        subtensor.substrate.create_signed_extrinsic.return_value,
+        extrinsic=subtensor.substrate.create_signed_extrinsic.return_value,
         wait_for_inclusion=True,
         wait_for_finalization=True,
     )
@@ -170,7 +170,7 @@ async def test_do_transfer_no_waiting(subtensor, mocker):
         call=subtensor.substrate.compose_call.return_value, keypair=fake_wallet.coldkey
     )
     subtensor.substrate.submit_extrinsic.assert_called_once_with(
-        subtensor.substrate.create_signed_extrinsic.return_value,
+        extrinsic=subtensor.substrate.create_signed_extrinsic.return_value,
         wait_for_inclusion=False,
         wait_for_finalization=False,
     )
@@ -204,7 +204,7 @@ async def test_transfer_extrinsic_success(subtensor, mocker):
     mocked_get_balance = mocker.patch.object(
         subtensor,
         "get_balance",
-        return_value={fake_wallet.coldkeypub.ss58_address: 10000},
+        return_value=10000,
     )
     mocked_get_existential_deposit = mocker.patch.object(
         subtensor, "get_existential_deposit", return_value=1
@@ -269,7 +269,7 @@ async def test_transfer_extrinsic_call_successful_with_failed_response(
     mocked_get_balance = mocker.patch.object(
         subtensor,
         "get_balance",
-        return_value={fake_wallet.coldkeypub.ss58_address: 10000},
+        return_value=10000,
     )
     mocked_get_existential_deposit = mocker.patch.object(
         subtensor, "get_existential_deposit", return_value=1
@@ -333,9 +333,7 @@ async def test_transfer_extrinsic_insufficient_balance(subtensor, mocker):
     mocked_get_balance = mocker.patch.object(
         subtensor,
         "get_balance",
-        return_value={
-            fake_wallet.coldkeypub.ss58_address: 1000
-        },  # Insufficient balance
+        return_value=1000,  # Insufficient balance
     )
     mocked_get_existential_deposit = mocker.patch.object(
         subtensor, "get_existential_deposit", return_value=1
@@ -465,7 +463,7 @@ async def test_transfer_extrinsic_keep_alive_false_and_transfer_all_true(
     mocked_get_balance = mocker.patch.object(
         subtensor,
         "get_balance",
-        return_value={fake_wallet.coldkeypub.ss58_address: 1},
+        return_value=1,
     )
     mocked_get_existential_deposit = mocker.patch.object(
         subtensor, "get_existential_deposit", return_value=1

tests/unit_tests/extrinsics/test_async_root.py renamed to tests/unit_tests/extrinsics/asyncex/test_root.py

@@ -1,7 +1,7 @@
 import pytest
 from bittensor.core import async_subtensor
 from bittensor_wallet import Wallet
-from bittensor.core.extrinsics import async_weights
+from bittensor.core.extrinsics.asyncex import weights as async_weights
 
 
 @pytest.fixture(autouse=True)