tests: separate templates fixture

Author: zyzniewski-reef

tests/e2e_tests/conftest.py

@@ -49,13 +49,6 @@ def local_chain(request):
 
     # Pattern match indicates node is compiled and ready
     pattern = re.compile(r"Imported #1")
-
-    # install neuron templates
-    logging.info("downloading and installing neuron templates from github")
-    # commit with subnet-template-repo changes for rust wallet
-    templates_dir = clone_or_update_templates()
-    install_templates(templates_dir)
-
     timestamp = int(time.time())
 
     def wait_for_node_start(process, pattern):
@@ -102,8 +95,19 @@ def read_output():
     # Ensure the process has terminated
     process.wait()
 
-    # uninstall templates
+
+@pytest.fixture
+def templates():
+    logging.info("downloading and installing neuron templates from github")
+
+    templates_dir = clone_or_update_templates()
+
+    install_templates(templates_dir)
+
+    yield templates_dir
+
     logging.info("uninstalling neuron templates")
+
     uninstall_templates(template_path)
 
 

tests/e2e_tests/test_axon.py

@@ -4,14 +4,10 @@
 import pytest
 
 from bittensor.utils import networking
-from tests.e2e_tests.utils.e2e_test_utils import (
-    template_path,
-    templates_repo,
-)
 
 
 @pytest.mark.asyncio
-async def test_axon(subtensor, alice_wallet):
+async def test_axon(subtensor, templates, alice_wallet):
     """
     Test the Axon mechanism and successful registration on the network.
 
@@ -53,7 +49,7 @@ async def test_axon(subtensor, alice_wallet):
     cmd = " ".join(
         [
             f"{sys.executable}",
-            f'"{template_path}{templates_repo}/miner.py"',
+            f'"{templates}/miner.py"',
             "--netuid",
             str(netuid),
             "--subtensor.network",

tests/e2e_tests/test_dendrite.py

@@ -10,14 +10,10 @@
     sudo_set_hyperparameter_values,
     wait_epoch,
 )
-from tests.e2e_tests.utils.e2e_test_utils import (
-    template_path,
-    templates_repo,
-)
 
 
 @pytest.mark.asyncio
-async def test_dendrite(local_chain, subtensor, alice_wallet, bob_wallet):
+async def test_dendrite(local_chain, subtensor, templates, alice_wallet, bob_wallet):
     """
     Test the Dendrite mechanism
 
@@ -112,7 +108,7 @@ async def test_dendrite(local_chain, subtensor, alice_wallet, bob_wallet):
     cmd = " ".join(
         [
             f"{sys.executable}",
-            f'"{template_path}{templates_repo}/validator.py"',
+            f'"{templates}/validator.py"',
             "--netuid",
             str(netuid),
             "--subtensor.network",

tests/e2e_tests/test_incentive.py

@@ -7,14 +7,10 @@
     sudo_set_hyperparameter_values,
     wait_epoch,
 )
-from tests.e2e_tests.utils.e2e_test_utils import (
-    template_path,
-    templates_repo,
-)
 
 
 @pytest.mark.asyncio
-async def test_incentive(local_chain, subtensor, alice_wallet, bob_wallet):
+async def test_incentive(local_chain, subtensor, templates, alice_wallet, bob_wallet):
     """
     Test the incentive mechanism and interaction of miners/validators
 
@@ -75,7 +71,7 @@ async def test_incentive(local_chain, subtensor, alice_wallet, bob_wallet):
     cmd = " ".join(
         [
             f"{sys.executable}",
-            f'"{template_path}{templates_repo}/miner.py"',
+            f'"{templates}/miner.py"',
             "--netuid",
             str(netuid),
             "--subtensor.network",
@@ -104,7 +100,7 @@ async def test_incentive(local_chain, subtensor, alice_wallet, bob_wallet):
     cmd = " ".join(
         [
             f"{sys.executable}",
-            f'"{template_path}{templates_repo}/validator.py"',
+            f'"{templates}/validator.py"',
             "--netuid",
             str(netuid),
             "--subtensor.network",

tests/e2e_tests/test_root_set_weights.py

@@ -6,10 +6,6 @@
     wait_epoch,
     sudo_set_hyperparameter_values,
 )
-from tests.e2e_tests.utils.e2e_test_utils import (
-    template_path,
-    templates_repo,
-)
 
 FAST_BLOCKS_SPEEDUP_FACTOR = 5
 
@@ -34,7 +30,7 @@
 
 
 @pytest.mark.asyncio
-async def test_root_reg_hyperparams(local_chain, subtensor, alice_wallet, bob_wallet):
+async def test_root_reg_hyperparams(local_chain, subtensor, templates, alice_wallet, bob_wallet):
     """
     Test root weights and hyperparameters in the Subtensor network.
 
@@ -86,7 +82,7 @@ async def test_root_reg_hyperparams(local_chain, subtensor, alice_wallet, bob_wa
     cmd = " ".join(
         [
             f"{sys.executable}",
-            f'"{template_path}{templates_repo}/validator.py"',
+            f'"{templates}/validator.py"',
             "--netuid",
             str(netuid),
             "--subtensor.network",

tests/e2e_tests/test_subtensor_functions.py

@@ -8,10 +8,6 @@
     sudo_set_admin_utils,
     wait_epoch,
 )
-from tests.e2e_tests.utils.e2e_test_utils import (
-    template_path,
-    templates_repo,
-)
 
 """
 Verifies:
@@ -35,7 +31,7 @@
 
 
 @pytest.mark.asyncio
-async def test_subtensor_extrinsics(subtensor, alice_wallet, bob_wallet):
+async def test_subtensor_extrinsics(subtensor, templates, alice_wallet, bob_wallet):
     """
     Tests subtensor extrinsics
 
@@ -163,7 +159,7 @@ async def test_subtensor_extrinsics(subtensor, alice_wallet, bob_wallet):
     cmd = " ".join(
         [
             f"{sys.executable}",
-            f'"{template_path}{templates_repo}/validator.py"',
+            f'"{templates}/validator.py"',
             "--netuid",
             str(netuid),
             "--subtensor.network",

tests/e2e_tests/utils/e2e_test_utils.py

@@ -73,7 +73,7 @@ def clone_or_update_templates(specific_commit=None):
 
     os.chdir(cwd)
 
-    return install_dir + templates_repo + "/"
+    return install_dir + templates_repo
 
 
 def install_templates(install_dir):