imporve tests

broadinstitute · Jul 26, 2023 · ec236a1 · ec236a1
1 parent da6f005
commit ec236a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
@@ -77,7 +77,7 @@ jobs:
         run: |
           export DATASETS_DIR=./hail_search/fixtures
           coverage run --source="./hail_search" --omit="./hail_search/__main__.py","./hail_search/test_utils.py" -m pytest hail_search/
-          coverage report --fail-under=98
+          coverage report --fail-under=99
 
   nodejs:
     runs-on: ubuntu-latest

diff --git a/hail_search/search.py b/hail_search/search.py
@@ -7,12 +7,9 @@ def search_hail_backend(request):
     data_types = list(sample_data.keys())
     single_data_type = data_types[0] if len(data_types) == 1 else None
 
-    if single_data_type:
-        sample_data = sample_data[single_data_type]
-        data_type = single_data_type
-        query_cls = QUERY_CLASS_MAP[single_data_type]
-    else:
-        raise NotImplementedError
+    sample_data = sample_data[single_data_type]
+    data_type = single_data_type
+    query_cls = QUERY_CLASS_MAP[single_data_type]
 
     query = query_cls(data_type, sample_data=sample_data, **request)
     return query.search()