generated from dopt-python/py311
110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from dopt_sensor_anomalies import _find_paths
|
|
|
|
# @pytest.fixture(scope="session", autouse=True)
|
|
# def setup_temp_dir(tmp_path_factory):
|
|
# tmp_dir = tmp_path_factory.mktemp("root")
|
|
# folder_structure = "lib/folder"
|
|
# pth = tmp_dir / folder_structure
|
|
# pth.mkdir(parents=True, exist_ok=True)
|
|
# # models
|
|
# folder_models = "lib/models"
|
|
# pth_models = tmp_dir / folder_models
|
|
# pth_models.mkdir(parents=True, exist_ok=True)
|
|
# _root_models = (Path(__file__).parent / "_models").glob("*.pth")
|
|
# for model in _root_models:
|
|
# dst = pth_models / model.name
|
|
# shutil.copy(model, dst)
|
|
|
|
# with patch("dopt_sensor_anomalies._find_paths.LIB_ROOT_PATH", pth):
|
|
# yield
|
|
|
|
|
|
@pytest.fixture()
|
|
def temp_model_folder_empty(tmp_path_factory) -> Path:
|
|
return tmp_path_factory.mktemp("empty")
|
|
|
|
|
|
@pytest.fixture()
|
|
def temp_model_folder_full(tmp_path_factory) -> Path:
|
|
folder = tmp_path_factory.mktemp("full")
|
|
left_hand_model = folder / "this_file_contains_the_left_hand_side_model.pth"
|
|
right_hand_model = folder / "this_file_contains_the_right_hand_side_model.pth"
|
|
left_hand_model.touch()
|
|
right_hand_model.touch()
|
|
return folder
|
|
|
|
|
|
@pytest.fixture()
|
|
def temp_model_folder_only_left(tmp_path_factory) -> Path:
|
|
folder = tmp_path_factory.mktemp("only_left")
|
|
left_hand_model = folder / "this_file_contains_the_left_hand_side_model.pth"
|
|
left_hand_model.touch()
|
|
return folder
|
|
|
|
|
|
@pytest.fixture()
|
|
def temp_model_folder_only_right(tmp_path_factory) -> Path:
|
|
folder = tmp_path_factory.mktemp("only_right")
|
|
right_hand_model = folder / "this_file_contains_the_right_hand_side_model.pth"
|
|
right_hand_model.touch()
|
|
return folder
|
|
|
|
|
|
@patch("dopt_sensor_anomalies._find_paths.STOP_FOLDER_NAME", "not-found")
|
|
def test_get_model_folder_Fail_NotFound():
|
|
with pytest.raises(FileNotFoundError):
|
|
_ = _find_paths.get_model_folder()
|
|
|
|
|
|
@patch("dopt_sensor_anomalies._find_paths.STOP_FOLDER_NAME", "lib")
|
|
def test_get_model_folder_Success():
|
|
ret = _find_paths.get_model_folder()
|
|
assert ret is not None
|
|
assert ret.name == _find_paths.MODEL_FOLDER_NAME
|
|
|
|
|
|
def test_get_detection_models_FailEmptyDir(temp_model_folder_empty):
|
|
with pytest.raises(ValueError):
|
|
_ = _find_paths.get_detection_models(temp_model_folder_empty)
|
|
|
|
|
|
def test_get_detection_models_FailOnlyLeft(temp_model_folder_only_left):
|
|
with pytest.raises(ValueError):
|
|
_ = _find_paths.get_detection_models(temp_model_folder_only_left)
|
|
|
|
|
|
def test_get_detection_models_FailOnlyRight(temp_model_folder_only_right):
|
|
with pytest.raises(ValueError):
|
|
_ = _find_paths.get_detection_models(temp_model_folder_only_right)
|
|
|
|
|
|
def test_get_detection_models_FailTooManyLeft(temp_model_folder_full):
|
|
right_hand_model = (
|
|
temp_model_folder_full / "this_file_contains_the_left_hand_side_model2.pth"
|
|
)
|
|
right_hand_model.touch()
|
|
|
|
with pytest.raises(ValueError):
|
|
_ = _find_paths.get_detection_models(temp_model_folder_full)
|
|
|
|
|
|
def test_get_detection_models_FailTooManyRight(temp_model_folder_full):
|
|
right_hand_model = (
|
|
temp_model_folder_full / "this_file_contains_the_right_hand_side_model2.pth"
|
|
)
|
|
right_hand_model.touch()
|
|
|
|
with pytest.raises(ValueError):
|
|
_ = _find_paths.get_detection_models(temp_model_folder_full)
|
|
|
|
|
|
def test_get_detection_models_Success(temp_model_folder_full):
|
|
models = _find_paths.get_detection_models(temp_model_folder_full)
|
|
assert "left_hand" in models["left"].name
|
|
assert "right_hand" in models["right"].name
|