mirror of
https://github.com/macaodha/batdetect2.git
synced 2025-06-29 14:41:58 +02:00
25 lines
720 B
Python
25 lines
720 B
Python
"""Test suite for loading batdetect annotations."""
|
|
|
|
from pathlib import Path
|
|
|
|
from soundevent import data
|
|
|
|
from batdetect2.data import BatDetect2FilesAnnotations, load_annotated_dataset
|
|
|
|
ROOT_DIR = Path(__file__).parent.parent.parent
|
|
|
|
|
|
def test_load_example_annotation_project():
|
|
path = ROOT_DIR / "example_data" / "anns"
|
|
audio_dir = ROOT_DIR / "example_data" / "audio"
|
|
annotation_set = load_annotated_dataset(
|
|
BatDetect2FilesAnnotations(
|
|
name="test",
|
|
audio_dir=audio_dir,
|
|
annotations_dir=path,
|
|
)
|
|
)
|
|
assert isinstance(annotation_set, data.AnnotationSet)
|
|
assert annotation_set.name == "test"
|
|
assert len(annotation_set.clip_annotations) == 3
|