mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-04-19 08:55:15 +00:00
31 lines
861 B
Python
31 lines
861 B
Python
|
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
|
||
|
|
)
|