def test_check_header_1(): path = os.path.join(test_dir_path, "test_datasets") a_dataset = "sample_table_small.csv" with io.open(os.path.expanduser(os.path.join(path, a_dataset)), encoding="utf8") as f: header = next(unicode_csv_reader(f)) assert header == ["id", "left_a", "right_a", "label"] id_attr = "id" label_attr = "label" left_prefix = "left" right_prefix = "right" _check_header(header, id_attr, left_prefix, right_prefix, label_attr, [])
def test_check_header_5(self): path = os.path.join(test_dir_path, 'test_datasets') a_dataset = 'sample_table_small.csv' with io.open(os.path.expanduser(os.path.join(path, a_dataset)), encoding="utf8") as f: header = next(unicode_csv_reader(f)) self.assertEqual(header, ['id', 'left_a', 'right_a', 'label']) id_attr = 'id' label_attr = '' left_prefix = 'left' right_prefix = 'right' _check_header(header, id_attr, left_prefix, right_prefix, label_attr, [])