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.
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
"""Unit tests for the WinCLIP torch model."""
|
|
|
|
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import pytest
|
|
import torch
|
|
|
|
from anomalib.models.image.winclip.torch_model import WinClipModel
|
|
|
|
|
|
class TestSetupWinClipModel:
|
|
"""Test the WinCLIP torch model."""
|
|
|
|
@staticmethod
|
|
def test_zero_shot_from_init() -> None:
|
|
"""Test WinCLIP initialization from init method in zero-shot mode."""
|
|
model = WinClipModel(class_name="item")
|
|
assert model.k_shot == 0
|
|
assert getattr(model, "text_embeddings", None) is not None
|
|
|
|
@staticmethod
|
|
def test_zero_shot_from_setup() -> None:
|
|
"""Test WinCLIP initialization from setup method in zero-shot mode."""
|
|
model = WinClipModel()
|
|
model.setup(class_name="item")
|
|
assert model.k_shot == 0
|
|
assert getattr(model, "text_embeddings", None) is not None
|
|
|
|
@staticmethod
|
|
@pytest.mark.parametrize("apply_transform", [True, False])
|
|
def test_few_shot_from_init(apply_transform: bool) -> None:
|
|
"""Test WinCLIP initialization from init in few-shot mode."""
|
|
ref_images = torch.rand(2, 3, 240, 240)
|
|
model = WinClipModel(class_name="item", reference_images=ref_images, apply_transform=apply_transform)
|
|
assert model.k_shot == 2
|
|
assert getattr(model, "text_embeddings", None) is not None
|
|
assert getattr(model, "visual_embeddings", None) is not None
|
|
|
|
@staticmethod
|
|
@pytest.mark.parametrize("apply_transform", [True, False])
|
|
def test_few_shot_from_setup(apply_transform: bool) -> None:
|
|
"""Test WinCLIP initialization from setup method in few-shot mode."""
|
|
ref_images = torch.rand(2, 3, 240, 240)
|
|
model = WinClipModel(apply_transform=apply_transform)
|
|
model.setup(class_name="item", reference_images=ref_images)
|
|
assert model.k_shot == 2
|
|
assert getattr(model, "text_embeddings", None) is not None
|
|
assert getattr(model, "visual_embeddings", None) is not None
|
|
|
|
@staticmethod
|
|
def test_raises_error_when_not_initialized() -> None:
|
|
"""Test if an error is raised when trying to access un-initialized attributes."""
|
|
model = WinClipModel()
|
|
with pytest.raises(RuntimeError):
|
|
_ = model.text_embeddings
|
|
with pytest.raises(RuntimeError):
|
|
_ = model.visual_embeddings
|
|
with pytest.raises(RuntimeError):
|
|
_ = model.patch_embeddings
|