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)
