Improve tests logic for Part 3 of subtensor refactoring

tests/unit_tests/test_subtensor.py

@@ -25,7 +25,7 @@
 
 # Application
 import bittensor
-from bittensor.subtensor import subtensor as Subtensor, _logger, Balance
+from bittensor.subtensor import subtensor as Subtensor, _logger, Balance, U16_NORMALIZED_FLOAT, U64_NORMALIZED_FLOAT
 from bittensor import subtensor_module
 
 
@@ -402,6 +402,10 @@ def test_hyper_parameter_success_calls(
     # Prep
     subtensor._get_hyperparameter = mocker.MagicMock(return_value=value)
 
+    spy_u16_normalized_float = mocker.spy(subtensor_module, 'U16_NORMALIZED_FLOAT')
+    spy_u64_normalized_float = mocker.spy(subtensor_module, 'U64_NORMALIZED_FLOAT')
+    spy_balance_from_rao = mocker.spy(Balance, "from_rao")
+
     # Call
     subtensor_method = getattr(subtensor, method)
     result = subtensor_method(netuid=7, block=707)
@@ -413,6 +417,16 @@ def test_hyper_parameter_success_calls(
     # if we change the methods logic in the future we have to be make sure tha returned type is correct
     assert isinstance(result, expected_result_type)
 
+    # Special cases
+    if method in ['kappa', 'validator_logits_divergence', 'validator_exclude_quantile', 'max_weight_limit']:
+        spy_u16_normalized_float.assert_called_once()
+
+    if method in ['adjustment_alpha', 'bonds_moving_avg']:
+        spy_u64_normalized_float.assert_called_once()
+
+    if method in ['recycle']:
+        spy_balance_from_rao.assert_called_once()
+
 
 def test_blocks_since_last_update_success_calls(subtensor, mocker):
     """Tests the weights_rate_limit method to ensure it correctly fetches the LastUpdate hyperparameter."""
@@ -427,6 +441,7 @@ def test_blocks_since_last_update_success_calls(subtensor, mocker):
     result = subtensor.blocks_since_last_update(netuid=7, uid=uid)
 
     # Assertions
+    subtensor.get_current_block.assert_called_once()
     subtensor._get_hyperparameter.assert_called_once_with(
         param_name="LastUpdate", netuid=7
     )