def test_disambiguate_table_subject_column_only_case_1():
table = GenericTable()
table.table = CASE_1_TABLE
table.subject_column = 1
entities = disambiguate_table_subject_column_only(table)
import ipdb
ipdb.set_trace()
def get_additional_tables(self):
tables = []
subject_column_list = os.path.join(ADDITIONAL_DATA_DIR,
"subject_columns.csv")
id_list = self.load_csv(subject_column_list)
for (_id, subject_column) in id_list:
table_filename = os.path.join(ADDITIONAL_DATA_DIR, "tables", _id)
table = GenericTable(filename=table_filename, _id=_id)
table.init()
table.table = table.table[:int(ROWS_TO_ANALYZE)]
table.subject_column = int(subject_column)
tables.append(table)
return tables