fixed test_serotyping - KeyError: 'serovar_in_serovar_list' error

phac-nml · Nov 26, 2024 · a02725e · a02725e
1 parent 0eacf75
commit a02725e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_serotyping.py b/tests/test_serotyping.py
@@ -12,12 +12,12 @@ def test_serotyping(fasta_path):
     assert  sistr_json_results['antigenic_formula'] == "58:l,z13,z28:z6"
     assert  sistr_json_results['cgmlst_matching_alleles'] == 330
     assert  sistr_json_results['cgmlst_found_loci'] == 330
-    assert  sistr_json_results['serovar_in_serovar_list'] == "Y"
+    assert  sistr_json_results['predicted_serovar_in_list'] == "Y"
     assert  sistr_json_results['serovar'] == "II 58:l,z13,z28:z6"
 
 def test_noserovarlist_file(fasta_path):
     sys.argv[1:] = ["-i",fasta_path, "test", "--more-results", "--run-mash", "--qc", "-l", "no_file_exists.txt", "-vvv", "-o", "sistr_test_results"]
     main() 
     with open("sistr_test_results.json") as fp:
         sistr_json_results = json.load(fp)[0]
-    assert 'serovar_in_serovar_list' not in sistr_json_results.keys()       
+    assert 'predicted_serovar_in_list' not in sistr_json_results.keys()