diff --git a/test/conftest.py b/test/conftest.py index 4ce522a..c835c54 100644 --- a/test/conftest.py +++ b/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) diff --git a/test/fixtures/input.py b/test/fixtures/input.py index e69de29..b50274a 100644 --- a/test/fixtures/input.py +++ b/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)) \ No newline at end of file diff --git a/test/fixtures/pdf.py b/test/fixtures/pdf.py index e69de29..1cc4f77 100644 --- a/test/fixtures/pdf.py +++ b/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() \ No newline at end of file