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.
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
"""Unit tests - MVTec AD 2 Datamodule."""
|
|
|
|
# Copyright (C) 2025 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from torchvision.transforms.v2 import Resize
|
|
|
|
from anomalib.data import MVTecAD2
|
|
from anomalib.data.datasets.image.mvtecad2 import TestType
|
|
from tests.unit.data.datamodule.base.image import _TestAnomalibImageDatamodule
|
|
|
|
|
|
class TestMVTecAD2(_TestAnomalibImageDatamodule):
|
|
"""MVTec AD 2 Datamodule Unit Tests."""
|
|
|
|
@pytest.fixture()
|
|
@staticmethod
|
|
def datamodule(dataset_path: Path) -> MVTecAD2:
|
|
"""Create and return a MVTec AD 2 datamodule."""
|
|
_datamodule = MVTecAD2(
|
|
root=dataset_path / "mvtecad2",
|
|
category="dummy",
|
|
train_batch_size=4,
|
|
eval_batch_size=4,
|
|
augmentations=Resize((256, 256)),
|
|
)
|
|
_datamodule.setup()
|
|
|
|
return _datamodule
|
|
|
|
@pytest.fixture()
|
|
@staticmethod
|
|
def fxt_data_config_path() -> str:
|
|
"""Return the path to the test data config."""
|
|
return "examples/configs/data/mvtecad2.yaml"
|
|
|
|
@staticmethod
|
|
def test_test_types(datamodule: MVTecAD2) -> None:
|
|
"""Test that the datamodule can handle different test types."""
|
|
# Test public test set
|
|
public_loader = datamodule.test_dataloader(test_type=TestType.PUBLIC)
|
|
assert public_loader is not None
|
|
batch = next(iter(public_loader))
|
|
assert batch.image.shape == (4, 3, 256, 256)
|
|
|
|
# Test private test set
|
|
private_loader = datamodule.test_dataloader(test_type=TestType.PRIVATE)
|
|
assert private_loader is not None
|
|
batch = next(iter(private_loader))
|
|
assert batch.image.shape == (4, 3, 256, 256)
|
|
|
|
# Test private mixed test set
|
|
mixed_loader = datamodule.test_dataloader(test_type=TestType.PRIVATE_MIXED)
|
|
assert mixed_loader is not None
|
|
batch = next(iter(mixed_loader))
|
|
assert batch.image.shape == (4, 3, 256, 256)
|
|
|
|
# Test invalid test type
|
|
with pytest.raises(ValueError, match="'invalid' is not a valid TestType"):
|
|
datamodule.test_dataloader(test_type="invalid")
|