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.
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
"""Unit Tests - Base Image Datamodules."""
|
|
|
|
# Copyright (C) 2023-2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import pytest
|
|
from torch.utils.data import DataLoader
|
|
|
|
from anomalib.data import AnomalibDataModule
|
|
|
|
|
|
class _TestAnomalibDataModule:
|
|
"""Base test class for ``AnomalibDataModule``.
|
|
|
|
This is a base class for testing the AnomalibDataModule. Since
|
|
``AnomalibDataModule`` has methods that are yet to be implemented, this base
|
|
test class is not meant to be used directly.
|
|
"""
|
|
|
|
@staticmethod
|
|
@pytest.mark.parametrize("subset", ["train", "val", "test"])
|
|
def test_datamodule_has_dataloader_attributes(datamodule: AnomalibDataModule, subset: str) -> None:
|
|
"""Test that the datamodule has the correct dataloader attributes."""
|
|
dataloader = f"{subset}_dataloader"
|
|
assert hasattr(datamodule, dataloader)
|
|
assert isinstance(getattr(datamodule, dataloader)(), DataLoader)
|
|
|
|
@staticmethod
|
|
def test_datamodule_from_config(fxt_data_config_path: str) -> None:
|
|
# 1. Wrong file path:
|
|
with pytest.raises(FileNotFoundError):
|
|
AnomalibDataModule.from_config(config_path="wrong_configs.yaml")
|
|
|
|
# 2. Correct file path:
|
|
datamodule = AnomalibDataModule.from_config(config_path=fxt_data_config_path)
|
|
assert datamodule is not None
|
|
assert isinstance(datamodule, AnomalibDataModule)
|
|
|
|
# 3. Override batch_size & num_workers
|
|
override_kwargs = {"data.train_batch_size": 1, "data.num_workers": 1}
|
|
datamodule = AnomalibDataModule.from_config(config_path=fxt_data_config_path, **override_kwargs)
|
|
assert datamodule.train_batch_size == 1
|
|
assert datamodule.num_workers == 1
|