import json
from pathlib import Path
from unittest.mock import patch

import pytest
from matplotlib.testing.compare import compare_images

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")],
)
@patch("akkudoktoreos.visualize.output_dir", DIR_IMAGEDATA)
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)
    assert (
        compare_images(
            str(DIR_IMAGEDATA / fn_out),
            str(DIR_IMAGEDATA / fn_out_base),
            0,
        )
        is None
    )