diff --git a/tests/unit_tests/extrinsics/test_async_weights.py b/tests/unit_tests/extrinsics/test_async_weights.py index 3fac1f7736..da250afd94 100644 --- a/tests/unit_tests/extrinsics/test_async_weights.py +++ b/tests/unit_tests/extrinsics/test_async_weights.py @@ -278,53 +278,6 @@ async def test_set_weights_extrinsic_exception(subtensor, mocker): assert message == "Unexpected error" -@pytest.mark.asyncio -async def test_set_weights_extrinsic_if_use_torch(subtensor, mocker): - """Tests set_weights_extrinsic when use_torch is True.""" - # Preps - fake_wallet = mocker.Mock(autospec=Wallet) - fake_netuid = 1 - fake_uids = [1, 2, 3] - fake_weights = [0.1, 0.2, 0.7] - - mocked_use_torch = mocker.patch.object( - async_weights, "use_torch", return_value=True - ) - mocked_torch_tensor = mocker.patch.object( - async_weights.torch, "tensor", return_value=mocker.Mock() - ) - - mocked_do_set_weights = mocker.patch.object( - async_weights, "_do_set_weights", return_value=(False, "Test error message") - ) - mocked_convert_weights_and_uids_for_emit = mocker.patch.object( - async_weights.weight_utils, - "convert_weights_and_uids_for_emit", - return_value=(mocker.Mock(), mocker.Mock()), - ) - - # Call - result, message = await async_weights.set_weights_extrinsic( - subtensor=subtensor, - wallet=fake_wallet, - netuid=fake_netuid, - uids=fake_uids, - weights=fake_weights, - wait_for_inclusion=True, - wait_for_finalization=True, - ) - - # Asserts - mocked_do_set_weights.assert_called_once() - mocked_use_torch.assert_called_once() - mocked_convert_weights_and_uids_for_emit.assert_called() - mocked_torch_tensor.assert_called_with( - fake_weights, dtype=async_weights.torch.float32 - ) - assert result is False - assert message == "Test error message" - - @pytest.mark.asyncio async def test_do_commit_weights_success(subtensor, mocker): """Tests _do_commit_weights when the commit is successful.""" diff --git a/tests/unit_tests/extrinsics/test_commit_reveal.py b/tests/unit_tests/extrinsics/test_commit_reveal.py index 3e95129343..243ef7cd2d 100644 --- a/tests/unit_tests/extrinsics/test_commit_reveal.py +++ b/tests/unit_tests/extrinsics/test_commit_reveal.py @@ -150,81 +150,6 @@ def test_do_commit_reveal_v3_failure_due_to_error(mocker, subtensor): assert result == (False, "Formatted error") -def test_commit_reveal_v3_extrinsic_success_with_torch(mocker, subtensor, hyperparams): - """Test successful commit-reveal with torch tensors.""" - # Preps - fake_wallet = mocker.Mock(autospec=subtensor_module.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) - fake_commit_for_reveal = b"mock_commit_for_reveal" - fake_reveal_round = 1 - - # Mocks - mocker.patch.object(commit_reveal, "use_torch", return_value=True) - - mocked_uids = mocker.Mock() - mocked_weights = mocker.Mock() - mocked_convert_weights_and_uids_for_emit = mocker.patch.object( - commit_reveal, - "convert_weights_and_uids_for_emit", - return_value=(mocked_uids, mocked_weights), - ) - mocked_get_subnet_reveal_period_epochs = mocker.patch.object( - subtensor, "get_subnet_reveal_period_epochs" - ) - mocked_get_encrypted_commit = mocker.patch.object( - commit_reveal, - "get_encrypted_commit", - return_value=(fake_commit_for_reveal, fake_reveal_round), - ) - mock_do_commit_reveal_v3 = mocker.patch.object( - commit_reveal, "_do_commit_reveal_v3", return_value=(True, "Success") - ) - mock_block = mocker.patch.object(subtensor, "get_current_block", return_value=1) - mock_hyperparams = mocker.patch.object( - subtensor, - "get_subnet_hyperparameters", - return_value=hyperparams, - ) - - # Call - success, message = commit_reveal.commit_reveal_v3_extrinsic( - subtensor=subtensor, - wallet=fake_wallet, - netuid=fake_netuid, - uids=fake_uids, - weights=fake_weights, - wait_for_inclusion=True, - wait_for_finalization=True, - ) - - # Asserts - assert success is True - assert message == "reveal_round:1" - mocked_convert_weights_and_uids_for_emit.assert_called_once_with( - fake_uids, fake_weights - ) - mocked_get_encrypted_commit.assert_called_once_with( - uids=mocked_uids, - weights=mocked_weights, - subnet_reveal_period_epochs=mock_hyperparams.return_value.commit_reveal_weights_interval, - version_key=commit_reveal.version_as_int, - tempo=mock_hyperparams.return_value.tempo, - netuid=fake_netuid, - current_block=mock_block.return_value, - ) - mock_do_commit_reveal_v3.assert_called_once_with( - self=subtensor, - wallet=fake_wallet, - netuid=fake_netuid, - commit=fake_commit_for_reveal, - reveal_round=fake_reveal_round, - wait_for_inclusion=True, - wait_for_finalization=True, - ) - - def test_commit_reveal_v3_extrinsic_success_with_numpy(mocker, subtensor, hyperparams): """Test successful commit-reveal with numpy arrays.""" # Preps @@ -233,7 +158,6 @@ def test_commit_reveal_v3_extrinsic_success_with_numpy(mocker, subtensor, hyperp fake_uids = np.array([1, 2, 3], dtype=np.int64) fake_weights = np.array([0.1, 0.2, 0.7], dtype=np.float32) - mocker.patch.object(commit_reveal, "use_torch", return_value=False) mock_convert = mocker.patch.object( commit_reveal, "convert_weights_and_uids_for_emit", @@ -282,7 +206,6 @@ def test_commit_reveal_v3_extrinsic_response_false(mocker, subtensor, hyperparam fake_reveal_round = 1 # Mocks - mocker.patch.object(commit_reveal, "use_torch", return_value=True) mocker.patch.object( commit_reveal, "convert_weights_and_uids_for_emit",