def test_convert_to_bq_string(self): hpo_rdr_mapping_list = gen_ext.get_hpo_and_rdr_mappings() hpo_bq_list = [] for hpo in resources.hpo_csv(): hpo_bq_list.append(self.bq_string.format(hpo_name=hpo["hpo_id"])) hpo_bq_list.append('("{rdr}", "{ppi_pm}")'.format( rdr=gen_ext.RDR, ppi_pm=gen_ext.PPI_PM)) expected = ', '.join(hpo_bq_list) actual = gen_ext.convert_to_bq_string(hpo_rdr_mapping_list) self.assertEqual(len(actual), len(expected))
def test_convert_to_bq_string(self, mock_hpo_list): mock_hpo_list.return_value = self.hpo_list hpo_rdr_mapping_list = gen_ext.get_hpo_and_rdr_mappings() hpo_bq_list = [] for hpo in bq_utils.get_hpo_info(): hpo_bq_list.append(self.bq_string.format(hpo_name=hpo["hpo_id"])) hpo_bq_list.append(f'("{gen_ext.RDR}", "{gen_ext.PPI_PM}")') expected = ', '.join(hpo_bq_list) actual = gen_ext.convert_to_bq_string(hpo_rdr_mapping_list) self.assertEqual(len(actual), len(expected))
def test_create_populate_source_mapping_table(self): # pre-conditions mapping_list = gen_ext.get_hpo_and_rdr_mappings() expected = str(len(mapping_list)) # test num_rows_affected = gen_ext.create_and_populate_source_mapping_table( self.bq_project_id, self.bq_dataset_id) # post condition self.assertEqual(expected, num_rows_affected)
def test_get_hpo_and_rdr_mappings(self): expected = [hpo_dict["hpo_id"] for hpo_dict in self.hpo_list] + [gen_ext.RDR] hpo_rdr_mapping_list = gen_ext.get_hpo_and_rdr_mappings() self.assertEqual(len(hpo_rdr_mapping_list), len(expected)) for hpo_rdr_id in expected: self.assertIn(hpo_rdr_id, [hpo_item[0] for hpo_item in hpo_rdr_mapping_list]) for hpo_item in hpo_rdr_mapping_list: self.assertIn(hpo_item[0], expected) if hpo_item[0] == gen_ext.RDR: self.assertEqual(hpo_item[1], gen_ext.PPI_PM)