Moving pytest plugin into the package namespace (#30)

This is neater, and is possible now #29 is merged.
etianen · Jan 27, 2024 · 003a0f8 · 003a0f8
1 parent aa171e7
commit 003a0f8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 21 deletions.
diff --git a/_logot_pytest.py b/_logot_pytest.py
diff --git a/logot/_pytest.py b/logot/_pytest.py
@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from collections.abc import Generator
+
+import pytest
+
+from logot._logot import Logot
+
+
+@pytest.fixture()
+def logot() -> Generator[Logot, None, None]:
+    with Logot().capturing() as logot:
+        yield logot
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,7 +17,7 @@ classifiers = [
     "Topic :: System :: Logging",
     "Typing :: Typed",
 ]
-packages = [{ include = "logot" }, { include = "_logot_pytest.py" }]
+packages = [{ include = "logot" }]
 
 [tool.poetry.dependencies]
 python = "^3.8"
@@ -43,7 +43,7 @@ sphinx = { version = "7.2.6", markers = "python_version >= '3.12'" }
 sphinx-autobuild = { version = "*", markers = "python_version >= '3.12'" }
 
 [tool.poetry.plugins.pytest11]
-logot = "_logot_pytest"
+logot = "logot._pytest"
 
 [tool.coverage.run]
 source = ["logot", "tests"]