def test_from_csv(self): tableset1 = TableSet(self.tables.values(), self.tables.keys()) tableset2 = TableSet.from_csv('examples/tableset', self.column_names, self.column_types) self.assertSequenceEqual(tableset1.column_names, tableset2.column_names) self.assertSequenceEqual(tableset1.column_types, tableset2.column_types) self.assertEqual(len(tableset1), len(tableset2)) for name in ['table1', 'table2', 'table3']: self.assertEqual(len(tableset1[name].columns), len(tableset2[name].columns)) self.assertEqual(len(tableset1[name].rows), len(tableset2[name].rows)) self.assertSequenceEqual(tableset1[name].rows[0], tableset2[name].rows[0]) self.assertSequenceEqual(tableset1[name].rows[1], tableset2[name].rows[1]) self.assertSequenceEqual(tableset1[name].rows[2], tableset2[name].rows[2])
def test_from_csv(self): tableset1 = TableSet(self.tables.values(), self.tables.keys()) tableset2 = TableSet.from_csv('examples/tableset', self.column_names) self.assertSequenceEqual(tableset1.column_names, tableset2.column_names) self.assertSequenceEqual([type(t) for t in tableset1.column_types], [type(t) for t in tableset2.column_types]) self.assertEqual(len(tableset1), len(tableset2)) for name in ['table1', 'table2', 'table3']: self.assertEqual(len(tableset1[name].columns), len(tableset2[name].columns)) self.assertEqual(len(tableset1[name].rows), len(tableset2[name].rows)) self.assertSequenceEqual(tableset1[name].rows[0], tableset2[name].rows[0]) self.assertSequenceEqual(tableset1[name].rows[1], tableset2[name].rows[1]) self.assertSequenceEqual(tableset1[name].rows[2], tableset2[name].rows[2])
def test_from_csv(self): tableset1 = TableSet(self.tables.values(), self.tables.keys()) tableset2 = TableSet.from_csv("examples/tableset", self.column_names) self.assertSequenceEqual(tableset1.column_names, tableset2.column_names) self.assertSequenceEqual([type(t) for t in tableset1.column_types], [type(t) for t in tableset2.column_types]) self.assertEqual(len(tableset1), len(tableset2)) for name in ["table1", "table2", "table3"]: self.assertEqual(len(tableset1[name].columns), len(tableset2[name].columns)) self.assertEqual(len(tableset1[name].rows), len(tableset2[name].rows)) self.assertSequenceEqual(tableset1[name].rows[0], tableset2[name].rows[0]) self.assertSequenceEqual(tableset1[name].rows[1], tableset2[name].rows[1]) self.assertSequenceEqual(tableset1[name].rows[2], tableset2[name].rows[2])
def test_tableset_from_csv_invalid_dir(self): with self.assertRaises(IOError): TableSet.from_csv('quack')
def test_from_csv_column_types_not_equal(self): with self.assertRaises(ValueError): TableSet.from_csv('examples/tableset/type_error')
def test_tableset_from_csv_invalid_dir(self): with self.assertRaises(IOError): TableSet.from_csv('quack')
def test_from_csv_column_types_not_equal(self): with self.assertRaises(ValueError): TableSet.from_csv('examples/tableset/type_error')