fix unit tests

Author: Roman

tests/unit_tests/extrinsics/asyncex/test_commit_reveal.py

@@ -73,11 +73,12 @@ async def test_do_commit_reveal_v3_success(mocker, subtensor, fake_wallet):
     # Asserts
     mocked_compose_call.assert_awaited_once_with(
         call_module="SubtensorModule",
-        call_function="commit_crv3_weights",
+        call_function="commit_timelocked_weights",
         call_params={
             "netuid": fake_netuid,
             "commit": fake_commit,
             "reveal_round": fake_reveal_round,
+            "commit_reveal_version": 4,
         },
     )
     mocked_create_signed_extrinsic.assert_awaited_once_with(
@@ -132,11 +133,12 @@ async def test_do_commit_reveal_v3_failure_due_to_error(mocker, subtensor, fake_
     # Asserts
     mocked_compose_call.assert_awaited_once_with(
         call_module="SubtensorModule",
-        call_function="commit_crv3_weights",
+        call_function="commit_timelocked_weights",
         call_params={
             "netuid": fake_netuid,
             "commit": fake_commit,
             "reveal_round": fake_reveal_round,
+            "commit_reveal_version": 4,
         },
     )
     mocked_create_signed_extrinsic.assert_awaited_once_with(
@@ -220,7 +222,7 @@ async def test_commit_reveal_v3_extrinsic_success_with_torch(
         netuid=fake_netuid,
         current_block=mock_block.return_value["header"]["number"],
         block_time=12.0,
-        hotkey=fake_wallet.hotkey.ss58_address.encode(),
+        hotkey=fake_wallet.hotkey.public_key,
     )
     mock_do_commit_reveal_v3.assert_awaited_once_with(
         subtensor=subtensor,

tests/unit_tests/extrinsics/test_commit_reveal.py

@@ -70,11 +70,12 @@ def test_do_commit_reveal_v3_success(mocker, subtensor, fake_wallet):
     # Asserts
     mocked_compose_call.assert_called_once_with(
         call_module="SubtensorModule",
-        call_function="commit_crv3_weights",
+        call_function="commit_timelocked_weights",
         call_params={
             "netuid": fake_netuid,
             "commit": fake_commit,
             "reveal_round": fake_reveal_round,
+            "commit_reveal_version": 4,
         },
     )
     mocked_sign_and_send_extrinsic.assert_called_once_with(
@@ -123,11 +124,12 @@ def test_do_commit_reveal_v3_failure_due_to_error(mocker, subtensor, fake_wallet
     # Asserts
     mocked_compose_call.assert_called_once_with(
         call_module="SubtensorModule",
-        call_function="commit_crv3_weights",
+        call_function="commit_timelocked_weights",
         call_params={
             "netuid": fake_netuid,
             "commit": fake_commit,
             "reveal_round": fake_reveal_round,
+            "commit_reveal_version": 4,
         },
     )
     mocked_create_signed_extrinsic.assert_called_once_with(
@@ -205,7 +207,7 @@ def test_commit_reveal_v3_extrinsic_success_with_torch(
         netuid=fake_netuid,
         current_block=mock_block.return_value,
         block_time=12.0,
-        hotkey=fake_wallet.hotkey.ss58_address.encode(),
+        hotkey=fake_wallet.hotkey.public_key,
     )
     mock_do_commit_reveal_v3.assert_called_once_with(
         subtensor=subtensor,