Pytest plugins in their own file

Author: JamesVorder

generated/test_class.py

@@ -2,17 +2,14 @@
 import typer
 
 # for the test runner
-import time
 import pytest
 # ------------------
 
 import langroid as lr
 from langroid.utils.configuration import set_global, Settings
 from langroid.utils.logging import setup_colored_logging
 
-# import the empty generated code
-import generated.test_class
-# ------------------
+from pytest_plugins import ResultsCollector, SessionStartPlugin
 
 app = typer.Typer()
 setup_colored_logging()
@@ -43,49 +40,6 @@ def generate_first_attempt() -> None:
         _out.write(response.content)
 
 
-class ResultsCollector:
-    def __init__(self):
-        self.reports = []
-        self.collected = 0
-        self.exitcode = 0
-        self.passed = 0
-        self.failed = 0
-        self.xfailed = 0
-        self.skipped = 0
-        self.total_duration = 0
-
-    @pytest.hookimpl(hookwrapper=True)
-    def pytest_runtest_makereport(self, item, call):
-        outcome = yield
-        report = outcome.get_result()
-        if report.when == 'call':
-            self.reports.append(report)
-
-    def pytest_collection_modifyitems(self, items):
-        self.collected = len(items)
-
-    def pytest_terminal_summary(self, terminalreporter, exitstatus):
-        self.exitcode = exitstatus
-        self.passed = len(terminalreporter.stats.get('passed', []))
-        self.failed = len(terminalreporter.stats.get('failed', []))
-        self.xfailed = len(terminalreporter.stats.get('xfailed', []))
-        self.skipped = len(terminalreporter.stats.get('skipped', []))
-
-        self.total_duration = time.time() - terminalreporter._sessionstarttime
-
-
-class SessionStartPlugin:
-    """
-    The goal of this plugin is to allow us to run pytest multiple times
-    and have it pick up the changes we generate in `generated/test_class.py`
-    """
-    def pytest_sessionstart(self):
-        if globals().get('generated', None) is not None:
-            import importlib
-            print("Reloading generated.test_class module...")
-            importlib.reload(generated.test_class)
-
-
 def get_test_results() -> str:
     collector = ResultsCollector()
     setup = SessionStartPlugin()

main.py

@@ -0,0 +1,49 @@
+import pytest
+import time
+try:
+    import generated.test_class
+except ImportError:
+    pass  # Since this is a generated file, it sometimes doesn't exist.
+
+
+class ResultsCollector:
+    def __init__(self):
+        self.reports = []
+        self.collected = 0
+        self.exitcode = 0
+        self.passed = 0
+        self.failed = 0
+        self.xfailed = 0
+        self.skipped = 0
+        self.total_duration = 0
+
+    @pytest.hookimpl(hookwrapper=True)
+    def pytest_runtest_makereport(self, item, call):
+        outcome = yield
+        report = outcome.get_result()
+        if report.when == 'call':
+            self.reports.append(report)
+
+    def pytest_collection_modifyitems(self, items):
+        self.collected = len(items)
+
+    def pytest_terminal_summary(self, terminalreporter, exitstatus):
+        self.exitcode = exitstatus
+        self.passed = len(terminalreporter.stats.get('passed', []))
+        self.failed = len(terminalreporter.stats.get('failed', []))
+        self.xfailed = len(terminalreporter.stats.get('xfailed', []))
+        self.skipped = len(terminalreporter.stats.get('skipped', []))
+
+        self.total_duration = time.time() - terminalreporter._sessionstarttime
+
+
+class SessionStartPlugin:
+    """
+    The goal of this plugin is to allow us to run pytest multiple times
+    and have it pick up the changes we generate in `generated/test_class.py`
+    """
+    def pytest_sessionstart(self):
+        if globals().get('generated', None) is not None:
+            import importlib
+            print("Reloading generated.test_class module...")
+            importlib.reload(generated.test_class)