Fix tests

Author: thewhaleking

pyproject.toml

@@ -49,4 +49,4 @@ requires = ["setuptools~=70.0.0", "wheel"]
 build-backend = "setuptools.build_meta"
 
 [project.optional-dependencies]
-dev = ["pytest"]
+dev = ["pytest", "bittensor", "pytest-asyncio"]

tests/test_old_new.py

@@ -1,8 +1,7 @@
 import os
-import asyncio
 import bittensor as bt
+import pytest
 
-use_async = (os.getenv("USE_ASYNC") == "1") or False
 try:
     n = int(os.getenv("NUMBER_RUNS"))
 except TypeError:
@@ -16,39 +15,37 @@
 b_pre = 4920340
 b_post = 4920360
 
-if use_async:
-    st = bt.async_subtensor(network="local")
-
-    async def main():
-        async with st:
-            print("ss58 format:", st.substrate.ss58_format)
-            print("current block (async):", await st.block)
-            for i in range(n):
-                s0 = await st.get_stake_for_coldkey(coldkey, block=b_post + i)
-                print(f"at block {b_post + i}: {s0}")
-            for i in range(n):
-                s1 = (
-                    await st.query_subtensor(
-                        "TotalColdkeyStake", block=b_pre + i, params=[coldkey]
-                    )
-                ).value
-                print(f"at block {b_pre + i}: {s1}")
-            for i in range(n):
-                s2 = await st.get_stake_for_coldkey(coldkey, block=b_post + i)
-                print(f"at block {b_post + i}: {s2}")
-
-    asyncio.run(main())
-else:
-    st = bt.subtensor(network="local")
-    print("ss58 format:", st.substrate.ss58_format)
-    print("current block (sync):", st.block)
-    st = bt.subtensor(network="local")
-    for i in range(n):
-        s0 = st.get_stake_for_coldkey(coldkey, block=b_post + i)
-        print(f"at block {b_post + i}: {s0}")
-    for i in range(n):
-        s1 = st.query_subtensor("TotalColdkeyStake", b_pre + i, [coldkey]).value
-        print(f"at block {b_pre + i}: {s1}")
-    for i in range(n):
-        s2 = st.get_stake_for_coldkey(coldkey, block=b_post + i)
-        print(f"at block {b_post + i}: {s2}")
+
+@pytest.mark.asyncio
+async def test_async():
+    async with bt.async_subtensor("archive") as st:
+        print("ss58 format:", st.substrate.ss58_format)
+        print("current block (async):", await st.block)
+        for i in range(n):
+            s0 = await st.get_stake_for_coldkey(coldkey, block=b_post + i)
+            print(f"at block {b_post + i}: {s0}")
+        for i in range(n):
+            s1 = (
+                await st.query_subtensor(
+                    "TotalColdkeyStake", block=b_pre + i, params=[coldkey]
+                )
+            ).value
+            print(f"at block {b_pre + i}: {s1}")
+        for i in range(n):
+            s2 = await st.get_stake_for_coldkey(coldkey, block=b_post + i)
+            print(f"at block {b_post + i}: {s2}")
+
+
+def test_sync():
+    with bt.subtensor("archive") as st:
+        print("ss58 format:", st.substrate.ss58_format)
+        print("current block (sync):", st.block)
+        for i in range(n):
+            s0 = st.get_stake_for_coldkey(coldkey, block=b_post + i)
+            print(f"at block {b_post + i}: {s0}")
+        for i in range(n):
+            s1 = st.query_subtensor("TotalColdkeyStake", b_pre + i, [coldkey]).value
+            print(f"at block {b_pre + i}: {s1}")
+        for i in range(n):
+            s2 = st.get_stake_for_coldkey(coldkey, block=b_post + i)
+            print(f"at block {b_post + i}: {s2}")