feat: init commit

neotest_python/__init__.py

@@ -0,0 +1,40 @@
+import argparse
+import json
+from enum import Enum
+from typing import List
+
+
+class TestRunner(str, Enum):
+    PYTEST = "pytest"
+    UNITTEST = "unittest"
+
+
+def get_adapter(runner: TestRunner):
+    if runner == TestRunner.PYTEST:
+        from .pytest import PytestNeotestAdapter
+
+        return PytestNeotestAdapter()
+    elif runner == TestRunner.UNITTEST:
+        from .unittest import UnittestNeotestAdapter
+
+        return UnittestNeotestAdapter()
+    raise NotImplementedError(runner)
+
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--runner", required=True)
+parser.add_argument(
+    "--results-file",
+    dest="results_file",
+    required=True,
+    help="File to store result JSON in",
+)
+parser.add_argument("args", nargs="*")
+
+
+def main(argv: List[str]):
+    args = parser.parse_args(argv)
+    adapter = get_adapter(TestRunner(args.runner))
+    results = adapter.run(args.args)
+    with open(args.results_file, "w") as results_file:
+        json.dump(results, results_file)