feat: init commit

neotest_python/base.py

@@ -0,0 +1,42 @@
+from enum import Enum
+from typing import TYPE_CHECKING, Dict, List, Optional
+
+
+class NeotestResultStatus(str, Enum):
+    SKIPPED = "skipped"
+    PASSED = "passed"
+    FAILED = "failed"
+
+    def __gt__(self, other) -> bool:
+        members = list(self.__class__.__members__.values())
+        return members.index(self) > members.index(other)
+
+
+if TYPE_CHECKING:
+    from typing import TypedDict
+
+    class NeotestError(TypedDict):
+        message: str
+        line: Optional[int]
+
+    class NeotestResult(TypedDict):
+        short: Optional[str]
+        status: NeotestResultStatus
+        errors: Optional[List[NeotestError]]
+
+else:
+    NeotestError = Dict
+    NeotestResult = Dict
+
+
+class NeotestAdapter:
+    def update_result(
+        self, base: Optional[NeotestResult], update: NeotestResult
+    ) -> NeotestResult:
+        if not base:
+            return update
+        return {
+            "status": max(base["status"], update["status"]),
+            "errors": (base.get("errors") or []) + (update.get("errors") or []) or None,
+            "short": (base.get("short") or "") + (update.get("short") or ""),
+        }