lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 14:41:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix batdetect2 test

Browse Source
This commit is contained in:
mbsantiago 2025-04-22 09:08:29 +01:00
parent e38c446f59
commit c276230bff
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_data/test_batdetect.py
View File

@ -12,14 +12,13 @@ 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_annotated_dataset(
annotation_set = load_annotated_dataset(
BatDetect2FilesAnnotations(
name="test",
audio_dir=audio_dir,
annotations_dir=path,
)
)
assert isinstance(project, data.AnnotationProject)
assert project.name == "test"
assert len(project.clip_annotations) == 3
assert len(project.tasks) == 3
assert isinstance(annotation_set, data.AnnotationSet)
assert annotation_set.name == "test"
assert len(annotation_set.clip_annotations) == 3