mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 05:16:34 +00:00
Oleksandr Bezdieniezhnykh
243b69656b
- Modified `.gitignore` to reflect the new path for test results. - Updated `docker-compose.test.yml` to mount the correct test results directory. - Adjusted `Dockerfile.test` to set the `PYTHONPATH` and ensure test results are saved in the updated location. - Added `boto3` and `netron` to `requirements-test.txt` to support new functionalities. - Updated `pytest.ini` to include the new `pythonpath` for test discovery. These changes streamline the testing process and ensure compatibility with the updated directory structure.
78 lines
2.0 KiB
Python
78 lines
2.0 KiB
Python
import re
|
|
|
|
import pytest
|
|
|
|
from dto.annotationClass import AnnotationClass
|
|
|
|
|
|
def _name_lines_under_names(text):
|
|
lines = text.splitlines()
|
|
out = []
|
|
in_block = False
|
|
for line in lines:
|
|
s = line.strip()
|
|
if s == "names:":
|
|
in_block = True
|
|
continue
|
|
if s.startswith("nc:"):
|
|
break
|
|
if in_block and s.startswith("-"):
|
|
out.append(s)
|
|
return out
|
|
|
|
|
|
_PLACEHOLDER_RE = re.compile(r"^-\s+Class-\d+\s*$")
|
|
|
|
|
|
@pytest.fixture
|
|
def data_yaml_text(monkeypatch, tmp_path, fixture_classes_json):
|
|
import train
|
|
|
|
import constants as c
|
|
monkeypatch.setattr(c, "config", c.Config(dirs=c.DirsConfig(root=str(tmp_path))))
|
|
monkeypatch.setattr(train, "today_folder", "")
|
|
from pathlib import Path
|
|
Path(c.config.datasets_dir).mkdir(parents=True, exist_ok=True)
|
|
train.create_yaml()
|
|
return (Path(c.config.datasets_dir) / "data.yaml").read_text(encoding="utf-8")
|
|
|
|
|
|
def test_bt_cls_01_base_classes(fixture_classes_json):
|
|
# Act
|
|
d = AnnotationClass.read_json()
|
|
norm = {k: d[k] for k in range(17)}
|
|
# Assert
|
|
assert len(norm) == 17
|
|
assert len({v.id for v in norm.values()}) == 17
|
|
|
|
|
|
def test_bt_cls_02_weather_expansion(fixture_classes_json):
|
|
# Act
|
|
d = AnnotationClass.read_json()
|
|
# Assert
|
|
assert d[0].name == "ArmorVehicle"
|
|
assert d[20].name == "ArmorVehicle(Wint)"
|
|
assert d[40].name == "ArmorVehicle(Night)"
|
|
|
|
|
|
@pytest.mark.resource_limit
|
|
def test_bt_cls_03_yaml_generation(data_yaml_text):
|
|
# Arrange
|
|
text = data_yaml_text
|
|
# Act
|
|
names = _name_lines_under_names(text)
|
|
placeholders = [ln for ln in names if _PLACEHOLDER_RE.match(ln)]
|
|
named = [ln for ln in names if not _PLACEHOLDER_RE.match(ln)]
|
|
# Assert
|
|
assert len(names) == 80
|
|
assert len(placeholders) == 29
|
|
assert len(named) == 51
|
|
|
|
|
|
@pytest.mark.resource_limit
|
|
def test_rl_cls_01_total_class_count(data_yaml_text):
|
|
# Act
|
|
names = _name_lines_under_names(data_yaml_text)
|
|
# Assert
|
|
assert len(names) == 80
|