Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Avoid using py.path #207

Merged
merged 2 commits into from
Nov 21, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 9 additions & 4 deletions pytest_mpl/plugin.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,11 @@
ALL_IMAGE_FORMATS = RASTER_IMAGE_FORMATS + VECTOR_IMAGE_FORMATS


def _get_item_dir(item):
# .path is available starting from pytest 7, .fspath is for older versions.
return getattr(item, "path", Path(item.fspath)).parent


def _hash_file(in_stream):
"""
Hashes an already opened file.
Expand Down Expand Up @@ -445,19 +450,19 @@ def get_baseline_directory(self, item):
baseline_dir = compare.kwargs.get('baseline_dir', None)
if baseline_dir is None:
if self.baseline_dir is None:
baseline_dir = Path(item.fspath).parent / 'baseline'
baseline_dir = _get_item_dir(item) / 'baseline'
else:
if self.baseline_relative_dir:
# baseline dir is relative to the current test
baseline_dir = Path(item.fspath).parent / self.baseline_relative_dir
baseline_dir = _get_item_dir(item) / self.baseline_relative_dir
else:
# baseline dir is relative to where pytest was run
baseline_dir = self.baseline_dir

baseline_remote = (isinstance(baseline_dir, str) and # noqa
baseline_dir.startswith(('http://', 'https://')))
if not baseline_remote:
return Path(item.fspath).parent / baseline_dir
return _get_item_dir(item) / baseline_dir

return baseline_dir

Expand Down Expand Up @@ -686,7 +691,7 @@ def compare_image_to_hash_library(self, item, fig, result_dir, summary=None):
hash_library_filename = compare.kwargs.get("hash_library", None) or self.hash_library
if self._hash_library_from_cli: # for backwards compatibility
hash_library_filename = self.hash_library
hash_library_filename = (Path(item.fspath).parent / hash_library_filename).absolute()
hash_library_filename = _get_item_dir(item) / hash_library_filename
ConorMacBride marked this conversation as resolved.
Show resolved Hide resolved

if not Path(hash_library_filename).exists():
pytest.fail(f"Can't find hash library at path {hash_library_filename}")
Expand Down
4 changes: 2 additions & 2 deletions tests/test_default_tolerance.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,12 @@


@pytest.fixture(scope="module")
def baseline_image(tmpdir_factory):
def baseline_image(tmp_path_factory):
path = Path(__file__).parent / "baseline" / "2.0.x" / f"{TEST_NAME}.png"
image = Image.open(path)
draw = ImageDraw.Draw(image)
draw.rectangle(((0, 0), (100, 100)), fill="red")
output = Path(tmpdir_factory.mktemp("data").join(f"{TEST_NAME}.png"))
output = tmp_path_factory.mktemp("data") / f"{TEST_NAME}.png"
image.save(output)
return output

Expand Down
106 changes: 49 additions & 57 deletions tests/test_pytest_mpl.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,11 +120,10 @@ def test_fail():
"""


def test_fails(tmpdir):

test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w') as f:
f.write(TEST_FAILING)
def test_fails(tmp_path):
test_file = tmp_path / "test.py"
test_file.write_text(TEST_FAILING)
test_file = str(test_file)

# If we use --mpl, it should detect that the figure is wrong
code = call_pytest(['--mpl', test_file])
Expand All @@ -147,16 +146,15 @@ def test_output_dir():
"""


def test_output_dir(tmpdir):
test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w') as f:
f.write(TEST_OUTPUT_DIR)
def test_output_dir(tmp_path):
test_file = tmp_path / "test.py"
test_file.write_text(TEST_OUTPUT_DIR)

output_dir = tmpdir.join('test_output_dir')
output_dir = tmp_path / "test_output_dir"

# When we run the test, we should get output images where we specify
code = call_pytest([f'--mpl-results-path={output_dir}',
'--mpl', test_file])
'--mpl', str(test_file)])

assert code != 0
assert output_dir.exists()
Expand All @@ -175,13 +173,14 @@ def test_gen():
"""


def test_generate(tmpdir):
def test_generate(tmp_path):

test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w') as f:
f.write(TEST_GENERATE)
test_file = tmp_path / "test.py"
test_file.write_text(TEST_GENERATE)
test_file = str(test_file)

gen_dir = tmpdir.mkdir('spam').mkdir('egg').strpath
gen_dir = tmp_path / "spam" / "egg"
gen_dir.mkdir(parents=True)

# If we don't generate, the test will fail
assert_pytest_fails_with(['--mpl', test_file], 'Image file not found for comparison test')
Expand Down Expand Up @@ -301,11 +300,10 @@ def test_hash_fails():
"""


def test_hash_fails(tmpdir):

test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w', encoding='ascii') as f:
f.write(TEST_FAILING_HASH)
def test_hash_fails(tmp_path):
test_file = tmp_path / "test.py"
test_file.write_text(TEST_FAILING_HASH, encoding="ascii")
test_file = str(test_file)

# If we use --mpl, it should detect that the figure is wrong
output = assert_pytest_fails_with(['--mpl', test_file], "doesn't match hash FAIL in library")
Expand Down Expand Up @@ -340,11 +338,11 @@ def test_hash_fail_hybrid():


@pytest.mark.skipif(ftv != '261', reason="Incorrect freetype version for hash check")
def test_hash_fail_hybrid(tmpdir):
def test_hash_fail_hybrid(tmp_path):

test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w', encoding='ascii') as f:
f.write(TEST_FAILING_HYBRID)
test_file = tmp_path / "test.py"
test_file.write_text(TEST_FAILING_HYBRID, encoding="ascii")
test_file = str(test_file)

# Assert that image comparison runs and fails
output = assert_pytest_fails_with(['--mpl', test_file,
Expand Down Expand Up @@ -382,18 +380,18 @@ def test_hash_fails():


@pytest.mark.skipif(ftv != '261', reason="Incorrect freetype version for hash check")
def test_hash_fail_new_hashes(tmpdir):
def test_hash_fail_new_hashes(tmp_path):
# Check that the hash comparison fails even if a new hash file is requested
test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w', encoding='ascii') as f:
f.write(TEST_FAILING_NEW_HASH)
test_file = tmp_path / "test.py"
test_file.write_text(TEST_FAILING_NEW_HASH, encoding="ascii")
test_file = str(test_file)

# Assert that image comparison runs and fails
assert_pytest_fails_with(['--mpl', test_file,
f'--mpl-hash-library={fail_hash_library}'],
"doesn't match hash FAIL in library")

hash_file = tmpdir.join('new_hashes.json').strpath
hash_file = tmp_path / "new_hashes.json"
# Assert that image comparison runs and fails
assert_pytest_fails_with(['--mpl', test_file,
f'--mpl-hash-library={fail_hash_library}',
Expand All @@ -413,11 +411,10 @@ def test_hash_missing():
"""


def test_hash_missing(tmpdir):

test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w') as f:
f.write(TEST_MISSING_HASH)
def test_hash_missing(tmp_path):
test_file = tmp_path / "test.py"
test_file.write_text(TEST_MISSING_HASH)
test_file = str(test_file)

# Assert fails if hash library missing
assert_pytest_fails_with(['--mpl', test_file, '--mpl-hash-library=/not/a/path'],
Expand Down Expand Up @@ -450,31 +447,26 @@ def test_unmodified(): return plot()


@pytest.mark.skipif(not hash_library.exists(), reason="No hash library for this mpl version")
def test_results_always(tmpdir):
def test_results_always(tmp_path):
test_file = tmp_path / "test.py"
test_file.write_text(TEST_RESULTS_ALWAYS)
results_path = tmp_path / "results"
results_path.mkdir()

test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w') as f:
f.write(TEST_RESULTS_ALWAYS)
results_path = tmpdir.mkdir('results')

code = call_pytest(['--mpl', test_file, '--mpl-results-always',
code = call_pytest(['--mpl', str(test_file), '--mpl-results-always',
rf'--mpl-hash-library={hash_library}',
rf'--mpl-baseline-path={baseline_dir_abs}',
'--mpl-generate-summary=html,json,basic-html',
rf'--mpl-results-path={results_path.strpath}'])
rf'--mpl-results-path={results_path}'])
assert code == 0 # hashes correct, so all should pass

# assert files for interactive HTML exist
assert results_path.join('fig_comparison.html').exists()
assert results_path.join('styles.css').exists()
assert results_path.join('extra.js').exists()

comparison_file = results_path.join('fig_comparison_basic.html')
with open(comparison_file, 'r') as f:
html = f.read()
assert (results_path / "fig_comparison.html").exists()
assert (results_path / "styles.css").exists()
assert (results_path / "extra.js").exists()

json_file = results_path.join('results.json')
with open(json_file, 'r') as f:
html = (results_path / "fig_comparison_basic.html").read_text()
with (results_path / "results.json").open("r") as f:
json_results = json.load(f)

# each test, and which images should exist
Expand All @@ -495,7 +487,7 @@ def test_results_always(tmpdir):

for image_type in ['baseline', 'result-failed-diff', 'result']:
image = f'{test_name}/{image_type}.png'
image_exists = results_path.join(*image.split('/')).exists()
image_exists = (results_path / image).exists()
json_image_key = f"{image_type.split('-')[-1]}_image"
if image_type in exists: # assert image so pytest prints it on error
assert image and image_exists
Expand Down Expand Up @@ -554,11 +546,11 @@ def test_fails(self):
TEST_FAILING_CLASS_SETUP_METHOD,
TEST_FAILING_UNITTEST_TESTCASE,
])
def test_class_fail(code, tmpdir):
def test_class_fail(code, tmp_path):

test_file = tmpdir.join('test.py').strpath
with open(test_file, 'w') as f:
f.write(code)
test_file = tmp_path / "test.py"
test_file.write_text(code)
test_file = str(test_file)

# Assert fails if hash library missing
assert_pytest_fails_with(['--mpl', test_file, '--mpl-hash-library=/not/a/path'],
Expand Down
Loading