mirror of
https://github.com/hanzoai/anomalib.git
synced 2026-10-30 18:48:11 +00:00
zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
162 lines
7.1 KiB
Python
162 lines
7.1 KiB
Python
"""Tests for path utils."""
|
|
|
|
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from pathlib import Path
|
|
from tempfile import TemporaryDirectory
|
|
|
|
import pytest
|
|
|
|
from anomalib.data.utils.path import validate_path
|
|
from anomalib.utils.path import generate_output_filename
|
|
|
|
|
|
class TestValidatePath:
|
|
"""Tests for ``validate_path`` function."""
|
|
|
|
@staticmethod
|
|
def test_invalid_path_type() -> None:
|
|
"""Test ``validate_path`` raises TypeError for an invalid path type."""
|
|
with pytest.raises(TypeError, match=r"Expected str, bytes or os.PathLike object, not*"):
|
|
validate_path(123)
|
|
|
|
@staticmethod
|
|
def test_is_path_too_long() -> None:
|
|
"""Test ``validate_path`` raises ValueError for a path that is too long."""
|
|
with pytest.raises(ValueError, match=r"Path is too long: *"):
|
|
validate_path("/" * 1000)
|
|
|
|
@staticmethod
|
|
def test_contains_non_printable_characters() -> None:
|
|
"""Test ``validate_path`` raises ValueError for a path that contains non-printable characters."""
|
|
with pytest.raises(ValueError, match=r"Path contains non-printable characters: *"):
|
|
validate_path("/\x00")
|
|
|
|
@staticmethod
|
|
def test_existing_file_within_base_dir(dataset_path: Path) -> None:
|
|
"""Test ``validate_path`` returns the validated path for an existing file within the base directory."""
|
|
file_path = dataset_path / "mvtecad/dummy/train/good/000.png"
|
|
validated_path = validate_path(file_path, base_dir=dataset_path)
|
|
assert validated_path == file_path.resolve()
|
|
|
|
@staticmethod
|
|
def test_existing_directory_within_base_dir(dataset_path: Path) -> None:
|
|
"""Test ``validate_path`` returns the validated path for an existing directory within the base directory."""
|
|
directory_path = dataset_path / "mvtecad/dummy/train/good"
|
|
validated_path = validate_path(directory_path, base_dir=dataset_path)
|
|
assert validated_path == directory_path.resolve()
|
|
|
|
@staticmethod
|
|
def test_nonexistent_file(dataset_path: Path) -> None:
|
|
"""Test ``validate_path`` raises FileNotFoundError for a nonexistent file."""
|
|
with pytest.raises(FileNotFoundError):
|
|
validate_path(dataset_path / "nonexistent/file.png")
|
|
|
|
@staticmethod
|
|
def test_nonexistent_directory(dataset_path: Path) -> None:
|
|
"""Test ``validate_path`` raises FileNotFoundError for a nonexistent directory."""
|
|
with pytest.raises(FileNotFoundError):
|
|
validate_path(dataset_path / "nonexistent/directory")
|
|
|
|
@staticmethod
|
|
def test_no_read_permission() -> None:
|
|
"""Test ``validate_path`` raises PermissionError for a file without read permission."""
|
|
with TemporaryDirectory() as tmp_dir:
|
|
file_path = Path(tmp_dir) / "test.txt"
|
|
with file_path.open("w") as f:
|
|
f.write("test")
|
|
file_path.chmod(0o222) # Remove read permission
|
|
with pytest.raises(PermissionError, match=r"Read or execute permissions denied for the path:*"):
|
|
validate_path(file_path, base_dir=Path(tmp_dir))
|
|
|
|
@staticmethod
|
|
def test_no_read_execute_permission() -> None:
|
|
"""Test ``validate_path`` raises PermissionError for a directory without read and execute permission."""
|
|
with TemporaryDirectory() as tmp_dir:
|
|
Path(tmp_dir).chmod(0o222) # Remove read and execute permission
|
|
with pytest.raises(PermissionError, match=r"Read or execute permissions denied for the path:*"):
|
|
validate_path(tmp_dir, base_dir=Path(tmp_dir))
|
|
|
|
@staticmethod
|
|
def test_file_wrongsuffix() -> None:
|
|
"""Test ``validate_path`` raises ValueError for a file with wrong suffix."""
|
|
with pytest.raises(ValueError, match="Path extension is not accepted."):
|
|
validate_path("file.png", should_exist=False, extensions=(".json", ".txt"))
|
|
|
|
|
|
class TestGenerateOutputFilename:
|
|
"""Tests for ``generate_output_filename`` function."""
|
|
|
|
@staticmethod
|
|
def test_basic_mvtec_style() -> None:
|
|
"""Test basic MVTec-style dataset path handling."""
|
|
input_path = Path("/data/MVTecAD/bottle/test/broken_large/000.png")
|
|
output_path = Path("./results")
|
|
result = generate_output_filename(input_path, output_path, "MVTecAD", "bottle")
|
|
assert result == Path("./results/test/broken_large/000.png")
|
|
|
|
@staticmethod
|
|
def test_without_category() -> None:
|
|
"""Test path handling without category parameter."""
|
|
input_path = Path("/data/MVTecAD/bottle/test/broken_large/000.png")
|
|
output_path = Path("./results")
|
|
result = generate_output_filename(input_path, output_path, "MVTecAD")
|
|
assert result == Path("./results/bottle/test/broken_large/000.png")
|
|
|
|
@staticmethod
|
|
def test_folder_dataset() -> None:
|
|
"""Test handling of folder-based datasets."""
|
|
input_path = Path("/datasets/MyDataset/normal/image001.png")
|
|
output_path = Path("./output")
|
|
result = generate_output_filename(input_path, output_path, "MyDataset")
|
|
assert result == Path("./output/normal/image001.png")
|
|
|
|
@staticmethod
|
|
def test_custom_structure() -> None:
|
|
"""Test handling of custom directory structures."""
|
|
input_path = Path("/custom/path/MyData/category/split/image.png")
|
|
output_path = Path("./out")
|
|
result = generate_output_filename(input_path, output_path, "MyData")
|
|
assert result == Path("./out/category/split/image.png")
|
|
|
|
@staticmethod
|
|
def test_auto_detection() -> None:
|
|
"""Test auto-detection when dataset_name is not provided."""
|
|
input_path = Path("/any/folder/structure/normal/image.png")
|
|
output_path = Path("./out")
|
|
result = generate_output_filename(input_path, output_path)
|
|
assert result == Path("./out/normal/image.png")
|
|
|
|
@staticmethod
|
|
def test_mkdir_parameter() -> None:
|
|
"""Test mkdir parameter behavior."""
|
|
with TemporaryDirectory() as tmp_dir:
|
|
input_path = Path("/data/MVTecAD/bottle/test/000.png")
|
|
output_path = Path(tmp_dir) / "results"
|
|
|
|
# Test with mkdir=True (default)
|
|
result = generate_output_filename(input_path, output_path, "MVTecAD")
|
|
assert result.parent.exists()
|
|
|
|
# Test with mkdir=False
|
|
output_path = Path(tmp_dir) / "results2"
|
|
result = generate_output_filename(input_path, output_path, "MVTecAD", mkdir=False)
|
|
assert not result.parent.exists()
|
|
|
|
@staticmethod
|
|
def test_case_insensitive_matching() -> None:
|
|
"""Test case-insensitive matching of dataset and category names."""
|
|
input_path = Path("/data/mvtecad/BOTTLE/test/000.png")
|
|
output_path = Path("./results")
|
|
result = generate_output_filename(input_path, output_path, "MVTecAD", "bottle")
|
|
assert result == Path("./results/test/000.png")
|
|
|
|
@staticmethod
|
|
def test_relative_paths() -> None:
|
|
"""Test handling of relative input paths."""
|
|
input_path = Path("data/MVTecAD/bottle/test/000.png")
|
|
output_path = Path("./results")
|
|
result = generate_output_filename(input_path, output_path, "MVTecAD")
|
|
assert result == Path("./results/bottle/test/000.png")
|