fix: properly mock data_chain class methods

opentensor · Feb 26, 2025 · 7b23a26 · 7b23a26
1 parent 46c6f62
commit 7b23a26
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/unit_tests/test_async_subtensor.py b/tests/unit_tests/test_async_subtensor.py
@@ -389,8 +389,10 @@ async def test_get_delegates(subtensor, mocker, fake_result, response):
         autospec=subtensor.query_runtime_api, return_value=fake_result
     )
     subtensor.query_runtime_api = mocked_query_runtime_api
-    mocked_delegate_info_list_from_dicts = mocker.Mock()
-    async_subtensor.DelegateInfo.list_from_dicts = mocked_delegate_info_list_from_dicts
+    mocked_delegate_info_list_from_dicts = mocker.patch.object(
+        async_subtensor.DelegateInfo,
+        "list_from_dicts",
+    )
 
     # Call
     result = await subtensor.get_delegates(block_hash=None, reuse_block=False)