Skip to content

Commit

Permalink
chore: add some missing tests in annotation filters
Browse files Browse the repository at this point in the history
  • Loading branch information
acostapazo committed Feb 6, 2023
1 parent 9063195 commit 68b8ef4
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions tests/unit/foundations/annotations/test_annotations.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,28 @@ def test_should_check_annotations_statistics():
]


@pytest.mark.unit
def test_should_filter_by_ids():
from gradgpad import annotations

selected_annotations = annotations.get_annotations_from_ids(
["casia-fasd_test_release/1/6.avi"]
)

assert len(selected_annotations) == 1


@pytest.mark.unit
def test_should_filter_annotations_by_dataset():
from gradgpad import annotations

selected_annotations = annotations.get_annotations_filtered_by_datasets(
["replay-mobile"]
)

assert len(selected_annotations) == 1030


@pytest.fixture
def expected_pseudo_random_ids_values():
return [
Expand Down

0 comments on commit 68b8ef4

Please sign in to comment.