EOS/tests/test_visualize.py

import json
from pathlib import Path

import pytest
from matplotlib.testing.compare import compare_images

from akkudoktoreos.config.config import get_config
from akkudoktoreos.visualize import visualisiere_ergebnisse

DIR_TESTDATA = Path(__file__).parent / "testdata"
DIR_IMAGEDATA = DIR_TESTDATA / "images"


@pytest.mark.parametrize(
    "fn_in, fn_out, fn_out_base",
    [("visualize_input_1.json", "visualize_output_1.pdf", "visualize_base_output_1.pdf")],
)
def test_visualisiere_ergebnisse(fn_in, fn_out, fn_out_base):
    with open(DIR_TESTDATA / fn_in, "r") as f:
        input_data = json.load(f)
    visualisiere_ergebnisse(**input_data)

    config = get_config()
    output_dir = config.data_output_path
    output_dir.mkdir(parents=True, exist_ok=True)
    output_file = output_dir.joinpath(fn_out)

    assert output_file.is_file()
    assert (
        compare_images(
            str(output_file),
            str(DIR_IMAGEDATA / fn_out_base),
            0,
        )
        is None
    )