ch03: transform into a pytest plugin

cjolowicz · cjolowicz · commit 942d872a392e · 2024-05-19T13:33:04.000+02:00
diff --git a/chapter03-python-packages/pyproject.toml b/chapter03-python-packages/pyproject.toml
@@ -1,9 +1,10 @@
 [project]
 name = "random-wikipedia-article"
 version = "0.1"
+dependencies = ["pytest"]
 
-[project.scripts]
-random-wikipedia-article = "random_wikipedia_article:main"
+[project.entry-points.pytest11]
+random-wikipedia-article = "random_wikipedia_article"
 
 [build-system]
 requires = ["hatchling"]
diff --git a/chapter03-python-packages/src/random_wikipedia_article/__init__.py b/chapter03-python-packages/src/random_wikipedia_article/__init__.py
@@ -1,13 +1,12 @@
 import json
-import textwrap
 import urllib.request
 
+import pytest
+
 API_URL = "https://en.wikipedia.org/api/rest_v1/page/random/summary"
 
 
-def main():
+@pytest.fixture
+def random_wikipedia_article():
     with urllib.request.urlopen(API_URL) as response:
-        data = json.load(response)
-
-    print(data["title"], end="\n\n")
-    print(textwrap.fill(data["extract"]))
+        return json.load(response)
diff --git a/chapter03-python-packages/src/random_wikipedia_article/__main__.py b/chapter03-python-packages/src/random_wikipedia_article/__main__.py
