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')