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,
[])
