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.
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
"""Unit Tests - Tabular Datamodule."""
|
|
|
|
# Copyright (C) 2025 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
from torchvision.transforms.v2 import Resize
|
|
|
|
from anomalib.data import Folder, Tabular
|
|
from tests.unit.data.datamodule.base.image import _TestAnomalibImageDatamodule
|
|
|
|
|
|
class TestTabular(_TestAnomalibImageDatamodule):
|
|
"""Tabular Datamodule Unit Tests."""
|
|
|
|
@staticmethod
|
|
def get_samples_dataframe(dataset_path: Path) -> pd.DataFrame:
|
|
"""Create samples DataFrame using the Folder datamodule."""
|
|
_folder_datamodule = Folder(
|
|
name="dummy",
|
|
root=dataset_path / "mvtecad" / "dummy",
|
|
normal_dir="train/good",
|
|
abnormal_dir="test/bad",
|
|
normal_test_dir="test/good",
|
|
mask_dir="ground_truth/bad",
|
|
train_batch_size=4,
|
|
eval_batch_size=4,
|
|
num_workers=0,
|
|
)
|
|
_folder_datamodule.setup()
|
|
return pd.concat([
|
|
_folder_datamodule.train_data.samples,
|
|
_folder_datamodule.test_data.samples,
|
|
_folder_datamodule.val_data.samples,
|
|
])
|
|
|
|
@pytest.fixture(
|
|
params=[
|
|
None,
|
|
["label"],
|
|
["label_index"],
|
|
["split"],
|
|
["mask_path"],
|
|
],
|
|
)
|
|
@staticmethod
|
|
def columns_to_drop(request: pytest.FixtureRequest) -> list[str] | None:
|
|
"""Return the columns to be dropped from the samples dataframe."""
|
|
return request.param
|
|
|
|
@pytest.fixture()
|
|
@staticmethod
|
|
def datamodule(dataset_path: Path, columns_to_drop: list | None) -> Tabular:
|
|
"""Create and return a Tabular datamodule."""
|
|
_samples = TestTabular.get_samples_dataframe(dataset_path)
|
|
if columns_to_drop:
|
|
_samples = _samples.drop(columns_to_drop, axis="columns")
|
|
_datamodule = Tabular(
|
|
name="dummy",
|
|
samples=_samples,
|
|
train_batch_size=4,
|
|
eval_batch_size=4,
|
|
num_workers=0,
|
|
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/tabular.yaml"
|
|
|
|
|
|
class TestTabularFromFile(TestTabular):
|
|
"""Tabular Datamodule Unit Tests for alternative constructor.
|
|
|
|
Tests for the Datamodule creation from file.
|
|
"""
|
|
|
|
@pytest.fixture()
|
|
@staticmethod
|
|
def datamodule(dataset_path: Path) -> Tabular:
|
|
"""Create and return a Tabular datamodule."""
|
|
_samples = TestTabular.get_samples_dataframe(dataset_path)
|
|
with tempfile.NamedTemporaryFile(suffix=".csv") as samples_file:
|
|
_samples.to_csv(samples_file)
|
|
samples_file.seek(0)
|
|
|
|
_datamodule = Tabular.from_file(
|
|
name="dummy",
|
|
file_path=samples_file.name,
|
|
train_batch_size=4,
|
|
eval_batch_size=4,
|
|
num_workers=0,
|
|
augmentations=Resize((256, 256)),
|
|
)
|
|
_datamodule.setup()
|
|
|
|
return _datamodule
|