Update tests for new call signature

nadeemlab · Sep 30, 2024 · 6f3fdc9 · 6f3fdc9
1 parent fd7b7af
commit 6f3fdc9
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git a/test/apiserver/module_tests/test_counts_query_delegation_edge_cases.py b/test/apiserver/module_tests/test_counts_query_delegation_edge_cases.py
@@ -46,8 +46,6 @@ def main():
         phenotype_total = sum(
             phenotype_count['count'] for phenotype_count in response['counts']
         )
-        total = response['number_cells_in_study']
-        print(total)
         if phenotype_total != expected:
             raise ValueError(f'Got wrong number: {phenotype_total}, expected {expected}.')
 

diff --git a/test/ondemand/module_tests/get_class_counts.py b/test/ondemand/module_tests/get_class_counts.py
@@ -6,7 +6,7 @@
 
 
 def get_counts(study_name, positives, negatives):
-    return OnDemandRequester.get_counts_by_specimen(positives, negatives, study_name, 0, ())
+    return OnDemandRequester.get_counts_by_specimen(positives, negatives, study_name, ())
 
 
 def main():

diff --git a/test/ondemand/module_tests/test_edge_cases_few_markers.py b/test/ondemand/module_tests/test_edge_cases_few_markers.py
@@ -5,7 +5,7 @@
 
 def retrieve_case(case):
     study_name = 'Melanoma intralesional IL2'
-    counts = OnDemandRequester.get_counts_by_specimen(case[0], case[1], study_name, 0, ())
+    counts = OnDemandRequester.get_counts_by_specimen(case[0], case[1], study_name, ())
     total = sum(entry.count for entry in counts.counts)
     return total
 

diff --git a/test/ondemand/module_tests/test_single_signature_count_query.py b/test/ondemand/module_tests/test_single_signature_count_query.py
@@ -7,7 +7,7 @@
 
 def main():
     study_name = 'Melanoma intralesional IL2'
-    counts = OnDemandRequester.get_counts_by_specimen(['CD3'], ['CD8', 'CD20'], study_name, 0, ())
+    counts = OnDemandRequester.get_counts_by_specimen(['CD3'], ['CD8', 'CD20'], study_name, ())
 
     counts_json = json.dumps(counts.model_dump(), indent=4).rstrip()
     with open('module_tests/expected_counts_structured1.json', 'rt', encoding='utf-8') as file: