class DataTableTest(unittest.TestCase): def setUp(self): self.addCleanup(self.my_cleanup, ('cleanup executado')) self.table = DataTable('A') def my_cleanup(self, msg): print(msg) def test_get_name(self): self.assertEqual("A", self.table._get_name()) def test_set_name(self): self.table._set_name("aurelio") self.assertEqual("aurelio", self.table._get_name()) def test_delete_name(self): self.assertRaises(AttributeError, self.table._del_name) def test_add_column(self): #table = DataTable("Empreendimento") self.assertEqual(0, len(self.table._columns)) self.table.add_column('BId', 'bigint') self.assertEqual(1, len(self.table._columns)) self.table.add_column('value', 'numeric') self.assertEqual(2, len(self.table._columns)) self.table.add_column('desc', 'varchar') self.assertEqual(3, len(self.table._columns)) def test_add_column_invalid_type(self): a_table = DataTable('A') self.assertRaises(Exception, a_table.add_column, ('col', 'invalid')) def test_add_column_invalid_type_fail(self): #a_table = DataTable("Empreendimento") error = False try: self.table.add_column('col', 'invalid') except: error = True if not error: self.fail("Chamada não gerou erro, mas deveria")