mirror of
https://github.com/macaodha/batdetect2.git
synced 2025-06-29 14:41:58 +02:00
20 lines
615 B
Python
20 lines
615 B
Python
"""Test suite for loading batdetect annotations."""
|
|
|
|
from pathlib import Path
|
|
|
|
from soundevent import data
|
|
|
|
from batdetect2.compat.data import load_annotation_project_from_dir
|
|
|
|
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"
|
|
project = load_annotation_project_from_dir(path, audio_dir=audio_dir)
|
|
assert isinstance(project, data.AnnotationProject)
|
|
assert project.name == str(path)
|
|
assert len(project.clip_annotations) == 3
|
|
assert len(project.tasks) == 3
|