diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 8f2a410..8da28e8 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -13,7 +13,7 @@ jobs: strategy: matrix: os: [macOS-latest, ubuntu-latest] - python-version: ['3.7', '3.8', '3.9', '3.10'] # TODO '3.11-dev' waiting for scikit-learn compatibility + python-version: ['3.7', '3.8', '3.9', '3.10', '3.11'] runs-on: ${{matrix.os}} steps: - uses: actions/checkout@v3 diff --git a/.github/workflows/installation.yml b/.github/workflows/installation.yml index 7cf3957..4675cdd 100644 --- a/.github/workflows/installation.yml +++ b/.github/workflows/installation.yml @@ -8,7 +8,7 @@ jobs: strategy: matrix: os: [macOS-latest, ubuntu-latest] - python-version: ['3.6', '3.7', '3.8', '3.9', '3.10'] # TODO '3.11-dev' waiting for scikit-learn compatibility + python-version: ['3.6', '3.7', '3.8', '3.9', '3.10', '3.11'] runs-on: ${{matrix.os}} continue-on-error: true steps: diff --git a/.github/workflows/reproducible_research.yml b/.github/workflows/reproducible_research.yml index f6f90ce..e3a3c1c 100644 --- a/.github/workflows/reproducible_research.yml +++ b/.github/workflows/reproducible_research.yml @@ -13,7 +13,7 @@ jobs: strategy: matrix: os: [macOS-latest, ubuntu-latest] - python-version: [3.9] + python-version: ['3.11'] runs-on: ${{matrix.os}} steps: - uses: actions/checkout@v3 diff --git a/tests/unit/foundations/annotations/test_annotations.py b/tests/unit/foundations/annotations/test_annotations.py index fd53c6b..fd02fbe 100644 --- a/tests/unit/foundations/annotations/test_annotations.py +++ b/tests/unit/foundations/annotations/test_annotations.py @@ -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 [