refactoring: splitting conftest logic into submodules

test/conftest.py

@@ -1,15 +1,10 @@
 import logging
-import os
 
-import fpdf
-import numpy as np
 import pytest
 
 from image_prediction.info import Info
-from image_prediction.locations import TEST_DATA_DIR
 from image_prediction.pipeline import load_pipeline
 from image_prediction.utils import get_logger
-from test.utils.generation.pdf import add_image, pdf_stream
 
 pytest_plugins = [
     "test.fixtures.image",
@@ -34,32 +29,11 @@ def mute_logger():
     logger.setLevel(level)
 
 
-@pytest.fixture
-def input_batch(batch_size, input_size):
-    return np.random.random_sample(size=(batch_size, *input_size))
-
-
 @pytest.fixture
 def info_label_map():
     return Info
 
 
-@pytest.fixture
-def pdf(image_metadata_pairs):
-    pdf = fpdf.FPDF(unit="pt")
-
-    for pair in image_metadata_pairs:
-        add_image(pdf, pair)
-
-    return pdf_stream(pdf)
-
-
-@pytest.fixture
-def real_pdf():
-    with open(os.path.join(TEST_DATA_DIR, "f2dc689ca794fccb8cd38b95f2bf6ba9.pdf"), "rb") as f:
-        yield f.read()
-
-
 @pytest.fixture
 def pipeline():
     pipeline = load_pipeline(verbose=False)

test/fixtures/input.py

@@ -0,0 +1,7 @@
+import numpy as np
+import pytest
+
+
+@pytest.fixture
+def input_batch(batch_size, input_size):
+    return np.random.random_sample(size=(batch_size, *input_size))

test/fixtures/pdf.py

@@ -0,0 +1,23 @@
+import os
+
+import fpdf
+import pytest
+
+from image_prediction.locations import TEST_DATA_DIR
+from test.utils.generation.pdf import add_image, pdf_stream
+
+
+@pytest.fixture
+def pdf(image_metadata_pairs):
+    pdf = fpdf.FPDF(unit="pt")
+
+    for pair in image_metadata_pairs:
+        add_image(pdf, pair)
+
+    return pdf_stream(pdf)
+
+
+@pytest.fixture
+def real_pdf():
+    with open(os.path.join(TEST_DATA_DIR, "f2dc689ca794fccb8cd38b95f2bf6ba9.pdf"), "rb") as f:
+        yield f.read()