Added prints to the tests

Author: thewhaleking

tests/integration_tests/test_async_substrate_interface.py

@@ -16,6 +16,7 @@
 
 @pytest.mark.asyncio
 async def test_legacy_decoding():
+    print("Testing test_legacy_decoding")
     # roughly 4000 blocks before metadata v15 was added
     pre_metadata_v15_block = 3_010_611
 
@@ -39,10 +40,12 @@ async def test_legacy_decoding():
             block_hash=block_hash,
         )
         assert timestamp.value == 1716358476004
+    print("test_legacy_decoding succeeded")
 
 
 @pytest.mark.asyncio
 async def test_ss58_conversion():
+    print("Testing test_ss58_conversion")
     async with AsyncSubstrateInterface(
         LATENT_LITE_ENTRYPOINT, ss58_format=42, decode_ss58=False
     ) as substrate:
@@ -79,10 +82,12 @@ async def test_ss58_conversion():
             if len(value.value) > 0:
                 for decoded_key in value.value:
                     assert isinstance(decoded_key, str)
+    print("test_ss58_conversion succeeded")
 
 
 @pytest.mark.asyncio
 async def test_fully_exhaust_query_map():
+    print("Testing test_fully_exhaust_query_map")
     async with AsyncSubstrateInterface(LATENT_LITE_ENTRYPOINT) as substrate:
         block_hash = await substrate.get_chain_finalised_head()
         non_fully_exhauster_start = time.time()
@@ -121,10 +126,12 @@ async def test_fully_exhaust_query_map():
             fully_exhausted_records_count += 1
         assert fully_exhausted_records_count == initial_records_count_fully_exhaust
         assert initial_records_count_fully_exhaust == exhausted_records_count
+    print("test_fully_exhaust_query_map succeeded")
 
 
 @pytest.mark.asyncio
 async def test_get_events_proper_decoding():
+    print("Testing test_get_events_proper_decoding")
     # known block/hash pair that has the events we seek to decode
     block = 5846788
     block_hash = "0x0a1c45063a59b934bfee827caa25385e60d5ec1fd8566a58b5cc4affc4eec412"
@@ -137,10 +144,12 @@ async def test_get_events_proper_decoding():
             30,
             "0xa6b4e5c8241d60ece0c25056b19f7d21ae845269fc771ad46bf3e011865129a5",
         )
+    print("test_get_events_proper_decoding succeeded")
 
 
 @pytest.mark.asyncio
 async def test_query_multiple():
+    print("Testing test_query_multiple")
     block = 6153277
     cks = [
         "5FH9AQM4kqbkdC9jyV5FrdEWVYt41nkhFstop7Vhyfb9ZsXt",
@@ -155,21 +164,25 @@ async def test_query_multiple():
             storage_function="OwnedHotkeys",
             block_hash=block_hash,
         )
+    print("test_query_multiple succeeded")
 
 
 @pytest.mark.asyncio
 async def test_reconnection():
+    print("Testing test_reconnection")
     async with AsyncSubstrateInterface(
         ARCHIVE_ENTRYPOINT, ss58_format=42, retry_timeout=8.0
     ) as substrate:
         await asyncio.sleep(9)  # sleep for longer than the retry timeout
         bh = await substrate.get_chain_finalised_head()
         assert isinstance(bh, str)
         assert isinstance(await substrate.get_block_number(bh), int)
+    print("test_reconnection succeeded")
 
 
 @pytest.mark.asyncio
 async def test_query_map_with_odd_number_of_params():
+    print("Testing test_query_map_with_odd_number_of_params")
     async with AsyncSubstrateInterface(ARCHIVE_ENTRYPOINT, ss58_format=42) as substrate:
         qm = await substrate.query_map(
             "SubtensorModule",
@@ -179,10 +192,12 @@ async def test_query_map_with_odd_number_of_params():
         first_record = qm.records[0]
         assert len(first_record) == 2
         assert len(first_record[0]) == 4
+    print("test_query_map_with_odd_number_of_params succeeded")
 
 
 @pytest.mark.asyncio
 async def test_improved_reconnection():
+    print("Testing test_improved_reconnection")
     ws_logger_path = "/tmp/websockets-proxy-test"
     ws_logger = logging.getLogger("websockets.proxy")
     if os.path.exists(ws_logger_path):
@@ -236,10 +251,12 @@ async def test_improved_reconnection():
     shutdown_thread.start()
     shutdown_thread.join(timeout=5)
     server_thread.join(timeout=5)
+    print("test_improved_reconnection succeeded")
 
 
 @pytest.mark.asyncio
 async def test_get_payment_info():
+    print("Testing test_get_payment_info")
     alice_coldkey = bittensor_wallet.Keypair.create_from_uri("//Alice")
     bob_coldkey = bittensor_wallet.Keypair.create_from_uri("//Bob")
     async with AsyncSubstrateInterface(
@@ -275,3 +292,4 @@ async def test_get_payment_info():
         partial_fee_all_options = payment_info_all_options["partial_fee"]
         assert partial_fee_all_options > partial_fee_no_era
         assert partial_fee_all_options > partial_fee_era
+    print("test_get_payment_info succeeded")

tests/integration_tests/test_disk_cache.py

@@ -9,6 +9,7 @@
 
 @pytest.mark.asyncio
 async def test_disk_cache():
+    print("Testing test_disk_cache")
     entrypoint = "wss://entrypoint-finney.opentensor.ai:443"
     async with DiskCachedAsyncSubstrateInterface(entrypoint) as disk_cached_substrate:
         current_block = await disk_cached_substrate.get_block_number(None)
@@ -72,3 +73,4 @@ async def test_disk_cache():
     assert parent_block_hash == parent_block_hash_sync
     assert block_runtime_info == block_runtime_info_sync
     assert block_runtime_version_for == block_runtime_version_for_sync
+    print("test_disk_cache succeeded")

tests/integration_tests/test_substrate_interface.py

@@ -7,6 +7,7 @@
 
 
 def test_legacy_decoding():
+    print("Testing test_legacy_decoding")
     # roughly 4000 blocks before metadata v15 was added
     pre_metadata_v15_block = 3_010_611
 
@@ -30,9 +31,11 @@ def test_legacy_decoding():
             block_hash=block_hash,
         )
         assert timestamp.value == 1716358476004
+    print("test_legacy_decoding succeeded")
 
 
 def test_ss58_conversion():
+    print("Testing test_ss58_conversion")
     with SubstrateInterface(
         LATENT_LITE_ENTRYPOINT, ss58_format=42, decode_ss58=False
     ) as substrate:
@@ -69,9 +72,11 @@ def test_ss58_conversion():
             if len(value.value) > 0:
                 for decoded_key in value.value:
                     assert isinstance(decoded_key, str)
+    print("test_ss58_conversion succeeded")
 
 
 def test_get_events_proper_decoding():
+    print("Testing test_get_events_proper_decoding")
     # known block/hash pair that has the events we seek to decode
     block = 5846788
     block_hash = "0x0a1c45063a59b934bfee827caa25385e60d5ec1fd8566a58b5cc4affc4eec412"
@@ -84,9 +89,11 @@ def test_get_events_proper_decoding():
             30,
             "0xa6b4e5c8241d60ece0c25056b19f7d21ae845269fc771ad46bf3e011865129a5",
         )
+    print("test_get_events_proper_decoding succeeded")
 
 
 def test_query_multiple():
+    print("Testing test_query_multiple")
     block = 6153277
     cks = [
         "5FH9AQM4kqbkdC9jyV5FrdEWVYt41nkhFstop7Vhyfb9ZsXt",
@@ -101,9 +108,11 @@ def test_query_multiple():
             storage_function="OwnedHotkeys",
             block_hash=block_hash,
         )
+    print("test_query_multiple succeeded")
 
 
 def test_query_map_with_odd_number_of_params():
+    print("Testing test_query_map_with_odd_number_of_params")
     with SubstrateInterface(LATENT_LITE_ENTRYPOINT, ss58_format=42) as substrate:
         qm = substrate.query_map(
             "SubtensorModule",
@@ -113,9 +122,11 @@ def test_query_map_with_odd_number_of_params():
         first_record = qm.records[0]
         assert len(first_record) == 2
         assert len(first_record[0]) == 4
+    print("test_query_map_with_odd_number_of_params succeeded")
 
 
 def test_get_payment_info():
+    print("Testing test_get_payment_info")
     alice_coldkey = bittensor_wallet.Keypair.create_from_uri("//Alice")
     bob_coldkey = bittensor_wallet.Keypair.create_from_uri("//Bob")
     with SubstrateInterface(
@@ -151,3 +162,4 @@ def test_get_payment_info():
         partial_fee_all_options = payment_info_all_options["partial_fee"]
         assert partial_fee_all_options > partial_fee_no_era
         assert partial_fee_all_options > partial_fee_era
+    print("test_get_payment_info succeeded")

tests/unit_tests/asyncio_/test_substrate_interface.py

@@ -13,19 +13,22 @@
 @pytest.mark.asyncio
 async def test_invalid_url_raises_exception():
     """Test that invalid URI raises an InvalidURI exception."""
+    print("Testing test_invalid_url_raises_exception")
     async_substrate = AsyncSubstrateInterface("non_existent_entry_point")
     with pytest.raises(InvalidURI):
         await async_substrate.initialize()
 
     with pytest.raises(InvalidURI):
         async with AsyncSubstrateInterface(
             "non_existent_entry_point"
-        ) as async_substrate:
+        ) as _:
             pass
+    print("test_invalid_url_raises_exception succeeded")
 
 
 @pytest.mark.asyncio
 async def test_runtime_call(monkeypatch):
+    print("Testing test_runtime_call")
     substrate = AsyncSubstrateInterface("ws://localhost", _mock=True)
 
     fake_runtime = MagicMock()
@@ -96,10 +99,12 @@ async def test_runtime_call(monkeypatch):
     substrate.rpc_request.assert_any_call(
         "state_call", ["SubstrateApi_SubstrateMethod", "", None], runtime=ANY
     )
+    print("test_runtime_call succeeded")
 
 
 @pytest.mark.asyncio
 async def test_websocket_shutdown_timer():
+    print("Testing test_websocket_shutdown_timer")
     # using default ws shutdown timer of 5.0 seconds
     async with AsyncSubstrateInterface("wss://lite.sub.latent.to:443") as substrate:
         await substrate.get_chain_head()
@@ -115,10 +120,12 @@ async def test_websocket_shutdown_timer():
         await substrate.get_chain_head()
         await asyncio.sleep(6)  # same sleep time as before
         assert substrate.ws.state is State.OPEN  # connection should still be open
+    print("test_websocket_shutdown_timer succeeded")
 
 
 @pytest.mark.asyncio
 async def test_runtime_switching():
+    print("Testing test_runtime_switching")
     block = 6067945  # block where a runtime switch happens
     async with AsyncSubstrateInterface(
         ARCHIVE_ENTRYPOINT, ss58_format=42, chain_name="Bittensor"
@@ -133,3 +140,4 @@ async def test_runtime_switching():
         )
         assert one is not None
         assert two is not None
+    print("test_runtime_switching succeeded")

tests/unit_tests/sync/test_substrate_interface.py

@@ -7,6 +7,7 @@
 
 
 def test_runtime_call(monkeypatch):
+    print("Testing test_runtime_call")
     substrate = SubstrateInterface("ws://localhost", _mock=True)
     fake_runtime = MagicMock()
     fake_metadata_v15 = MagicMock()
@@ -75,9 +76,11 @@ def test_runtime_call(monkeypatch):
         "state_call", ["SubstrateApi_SubstrateMethod", "", None]
     )
     substrate.close()
+    print("test_runtime_call succeeded")
 
 
 def test_runtime_switching():
+    print("Testing test_runtime_switching")
     block = 6067945  # block where a runtime switch happens
     with SubstrateInterface(
         ARCHIVE_ENTRYPOINT, ss58_format=42, chain_name="Bittensor"
@@ -86,3 +89,4 @@ def test_runtime_switching():
         assert substrate.get_extrinsics(block_number=block - 20) is not None
         assert substrate.get_extrinsics(block_number=block) is not None
         assert substrate.get_extrinsics(block_number=block - 21) is not None
+    print("test_runtime_switching succeeded")