def test_res_score_6(self):
self.results['Bla'] = 'a'
self.results['Bla_ESBL'] = 'b'
self.results['Bla_Carb'] = 'c'
self.results['Col'] = 'd'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['resistance_score'], '3')
def test_res_counts_2(self):
self.results['AGly'] = 'a'
self.results['Flq'] = 'b;c'
self.results['Tet'] = 'd;e;f'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['num_resistance_classes'], '3')
self.assertEqual(summary_results['num_resistance_genes'], '6')
def test_res_counts_6(self):
"""
Mutations should add to the class count but not to the gene count (because that is for
acquired resistance genes).
"""
self.results['AGly'] = 'a'
self.results['Flq'] = 'b;ParC-80I'
self.results['Tet'] = 'd;e;f'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['num_resistance_classes'], '3')
self.assertEqual(summary_results['num_resistance_genes'], '5')
def test_res_counts_4(self):
"""
Omp genes should not add to the counts.
"""
self.results['AGly'] = 'a'
self.results['Flq'] = 'b;c'
self.results['Tet'] = 'd;e;f'
self.results['Omp'] = 'g;h'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['num_resistance_classes'], '3')
self.assertEqual(summary_results['num_resistance_genes'], '6')
def test_res_counts_3(self):
"""
Intrinsic Bla genes should not add to the counts.
"""
self.results['AGly'] = 'a'
self.results['Flq'] = 'b;c'
self.results['Tet'] = 'd;e;f'
self.results['Bla'] = 'g;SHV-3'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['num_resistance_classes'], '3')
self.assertEqual(summary_results['num_resistance_genes'], '7')
def test_res_counts_8(self):
"""
Truncations should add to the class count but not to the gene count (because that is for
acquired resistance genes).
"""
self.results['AGly'] = 'a'
self.results['Tet'] = 'b;c'
self.results['Col'] = 'd;PmrB-20%'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['num_resistance_classes'], '3')
self.assertEqual(summary_results['num_resistance_genes'], '4')
def get_all_results(self, contigs):
results = {'strain': get_strain_name(contigs)}
results.update(get_contig_stat_results(contigs))
results.update(get_species_results(contigs, self.data_dir))
kp_complex = is_kp_complex(results)
results.update(
get_chromosome_mlst_results(self.data_dir, contigs, kp_complex))
results.update(get_ybt_mlst_results(self.data_dir, contigs))
results.update(get_clb_mlst_results(self.data_dir, contigs))
results.update(get_iuc_mlst_results(self.data_dir, contigs))
results.update(get_iro_mlst_results(self.data_dir, contigs))
results.update(get_hypermucoidy_results(self.data_dir, contigs))
results.update(get_wzi_and_k_locus_results(self.data_dir, contigs))
results.update(
get_resistance_results(self.data_dir, contigs, self.args,
self.res_headers, kp_complex))
results.update(get_summary_results(results, self.res_headers))
return results
def test_nothing(self):
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['virulence_score'], '0')
self.assertEqual(summary_results['resistance_score'], '0')
self.assertEqual(summary_results['num_resistance_classes'], '0')
self.assertEqual(summary_results['num_resistance_genes'], '0')
def test_vir_score_7(self):
self.results['Aerobactin'] = 'a'
self.results['Colibactin'] = 'b'
self.results['Yersiniabactin'] = 'c'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['virulence_score'], '5')
def test_vir_score_4(self):
self.results['Aerobactin'] = 'a'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['virulence_score'], '3')
def test_vir_score_1(self):
self.results['Yersiniabactin'] = 'a'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['virulence_score'], '1')
def test_res_score_7(self):
self.results['Col'] = 'a'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['resistance_score'], '0')
def test_res_score_4(self):
self.results['Bla'] = 'a'
self.results['Bla_Carb'] = 'b'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['resistance_score'], '2')
def test_res_score_3(self):
self.results['Bla'] = 'a'
self.results['Bla_ESBL_inhR'] = 'b'
summary_results = get_summary_results(self.results, self.res_headers)
self.assertEqual(summary_results['resistance_score'], '1')
