1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
from helpers import pytester_path
PYFILE = (
"""
import matplotlib.pyplot as plt
import pytest
@pytest.mark.mpl_image_compare
def test_mpl():
fig, ax = plt.subplots()
ax.plot([1, 2, 3])
return fig
"""
)
def test_generate_baseline_images(pytester):
pytester.makepyfile(PYFILE)
baseline_dir = pytester_path(pytester) / "alternative_baseline"
result = pytester.runpytest(f"--mpl-generate-path={baseline_dir}")
result.assert_outcomes(skipped=1)
assert (baseline_dir / "test_mpl.png").exists()
def test_generate_baseline_hashes(pytester):
pytester.makepyfile(PYFILE)
path = pytester_path(pytester)
hash_library = path / "alternative_baseline" / "hash_library_1.json"
result = pytester.runpytest(
f"--mpl-generate-hash-library={hash_library}",
f"--mpl-results-path={path}",
)
result.assert_outcomes(failed=1) # this option enables --mpl
assert hash_library.exists()
assert (path / "test_generate_baseline_hashes.test_mpl" / "result.png").exists()
def test_generate_baseline_images_and_hashes(pytester):
pytester.makepyfile(PYFILE)
path = pytester_path(pytester)
baseline_dir = path / "alternative_baseline"
hash_library = path / "alternative_baseline" / "hash_library_1.json"
result = pytester.runpytest(
f"--mpl-generate-path={baseline_dir}",
f"--mpl-generate-hash-library={hash_library}",
)
result.assert_outcomes(skipped=1)
assert (baseline_dir / "test_mpl.png").exists()
assert hash_library.exists()
|
