From fbd0db27ebe4d9f710dcbc4ae2924936e25191c8 Mon Sep 17 00:00:00 2001 From: Christian Oertlin Date: Wed, 10 Apr 2024 15:00:39 +0200 Subject: [PATCH] add test for sample filter --- tests/database/filters/test_sample_filters.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/database/filters/test_sample_filters.py b/tests/database/filters/test_sample_filters.py index 3b4507c..7609c3c 100644 --- a/tests/database/filters/test_sample_filters.py +++ b/tests/database/filters/test_sample_filters.py @@ -8,6 +8,7 @@ filter_samples_having_comment, filter_samples_without_status, filter_samples_analysed_on_plate, + add_skip_and_limit, ) from genotype_api.database.models import Sample, Plate from genotype_api.database.store import Store @@ -127,3 +128,17 @@ def test_filter_samples_analysed_on_plate_none_provided( # THEN no sample is returned assert len(samples) == 2 + + +def test_add_skip_and_limit(base_store: Store, test_sample: Sample): + """Test add_skip_and_limit function.""" + + # GIVEN a store with two samples + + # WHEN adding skip and limit to the query + query: Query = base_store._get_query(Sample) + samples: list[Sample] = add_skip_and_limit(query, skip=0, limit=1).all() + + # THEN one SNP is returned + assert samples + assert len(samples) == 1