def get_taipan_tables(self): tables = [] id_list = self.load_csv(SUBJECT_COLUMN_LIST) id_list = map((lambda x: x[0]), id_list) for _id in id_list: table = Table(_id) table.init() table.table = table.table[:int(ROWS_TO_ANALYZE)] tables.append(table) return tables
def test_is_subject_column(): table = Table(TABLE_ID) table.init() assert table.is_subject_column(1) == False
def test_get_subject_column_not_exist(): table = Table(TABLE_ID) subject_column = table.get_subject_column("ifosjfoijdsf") assert subject_column == None
def test_get_subject_column(): table = Table(TABLE_ID) subject_column = table.get_subject_column(TABLE_ID) assert subject_column == 0
def test_get_properties(): table = Table(TABLE_ID) properties = table.get_properties(TABLE_ID) assert properties == PROPERTIES
def test_get_classes(): table = Table(TABLE_ID) classes = table.get_classes(TABLE_ID) assert classes == CLASSES
def test_get_data(): table = Table(TABLE_ID) table_data = table.get_data(TABLE_ID) assert array_equal(table_data, TABLE_DATA)