mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-05-23 06:41:53 +02:00
133 lines
3.7 KiB
Python
133 lines
3.7 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
from soundevent import data
|
|
|
|
from batdetect2.data.conditions import (
|
|
ClipAllOfConfig,
|
|
ClipAnyOfConfig,
|
|
ClipNotConfig,
|
|
HasAllTagsConfig,
|
|
HasAnyTagConfig,
|
|
HasTagConfig,
|
|
IdInListConfig,
|
|
RecordingSatisfiesConfig,
|
|
build_clip_annotation_condition,
|
|
)
|
|
|
|
|
|
def test_recording_satisfies_condition(
|
|
tmp_path: Path,
|
|
create_recording,
|
|
create_clip,
|
|
create_clip_annotation,
|
|
) -> None:
|
|
recording_a = create_recording(path=tmp_path / "a.wav")
|
|
recording_b = create_recording(path=tmp_path / "b.wav")
|
|
clip_a = create_clip(recording_a)
|
|
clip_b = create_clip(recording_b)
|
|
clip_annotation_a = create_clip_annotation(clip_a)
|
|
clip_annotation_b = create_clip_annotation(clip_b)
|
|
ids_path = tmp_path / "recording_ids.json"
|
|
ids_path.write_text(json.dumps([str(recording_a.uuid)]))
|
|
|
|
condition = build_clip_annotation_condition(
|
|
RecordingSatisfiesConfig(
|
|
condition=IdInListConfig(path=ids_path),
|
|
)
|
|
)
|
|
|
|
assert condition(clip_annotation_a)
|
|
assert not condition(clip_annotation_b)
|
|
|
|
|
|
def test_clip_id_in_list_condition(
|
|
tmp_path: Path,
|
|
create_recording,
|
|
create_clip,
|
|
create_clip_annotation,
|
|
) -> None:
|
|
recording_a = create_recording(path=tmp_path / "a.wav")
|
|
recording_b = create_recording(path=tmp_path / "b.wav")
|
|
clip_annotation_a = create_clip_annotation(create_clip(recording_a))
|
|
clip_annotation_b = create_clip_annotation(create_clip(recording_b))
|
|
ids_path = tmp_path / "clip_annotation_ids.json"
|
|
ids_path.write_text(json.dumps([str(clip_annotation_a.uuid)]))
|
|
|
|
condition = build_clip_annotation_condition(IdInListConfig(path=ids_path))
|
|
|
|
assert condition(clip_annotation_a)
|
|
assert not condition(clip_annotation_b)
|
|
|
|
|
|
def test_clip_has_tag_conditions(
|
|
tmp_path: Path,
|
|
create_recording,
|
|
create_clip,
|
|
create_clip_annotation,
|
|
) -> None:
|
|
reviewed = data.Tag(key="status", value="reviewed")
|
|
train = data.Tag(key="split", value="train")
|
|
val = data.Tag(key="split", value="val")
|
|
|
|
recording = create_recording(path=tmp_path / "rec.wav")
|
|
clip = create_clip(recording)
|
|
clip_annotation = create_clip_annotation(
|
|
clip,
|
|
clip_tags=[reviewed, train],
|
|
)
|
|
|
|
has_tag = build_clip_annotation_condition(HasTagConfig(tag=reviewed))
|
|
has_all = build_clip_annotation_condition(
|
|
HasAllTagsConfig(tags=[reviewed, train])
|
|
)
|
|
has_any = build_clip_annotation_condition(
|
|
HasAnyTagConfig(tags=[val, train])
|
|
)
|
|
|
|
assert has_tag(clip_annotation)
|
|
assert has_all(clip_annotation)
|
|
assert has_any(clip_annotation)
|
|
|
|
|
|
def test_clip_logical_conditions(
|
|
tmp_path: Path,
|
|
create_recording,
|
|
create_clip,
|
|
create_clip_annotation,
|
|
) -> None:
|
|
reviewed = data.Tag(key="status", value="reviewed")
|
|
train = data.Tag(key="split", value="train")
|
|
val = data.Tag(key="split", value="val")
|
|
|
|
recording = create_recording(path=tmp_path / "rec.wav")
|
|
clip = create_clip(recording)
|
|
clip_annotation = create_clip_annotation(
|
|
clip,
|
|
clip_tags=[reviewed, train],
|
|
)
|
|
|
|
all_condition = build_clip_annotation_condition(
|
|
ClipAllOfConfig(
|
|
conditions=[
|
|
HasTagConfig(tag=reviewed),
|
|
HasAnyTagConfig(tags=[train, val]),
|
|
]
|
|
)
|
|
)
|
|
any_condition = build_clip_annotation_condition(
|
|
ClipAnyOfConfig(
|
|
conditions=[
|
|
HasTagConfig(tag=val),
|
|
HasTagConfig(tag=reviewed),
|
|
]
|
|
)
|
|
)
|
|
not_condition = build_clip_annotation_condition(
|
|
ClipNotConfig(condition=HasTagConfig(tag=val))
|
|
)
|
|
|
|
assert all_condition(clip_annotation)
|
|
assert any_condition(clip_annotation)
|
|
assert not_condition(clip_annotation)
|