diff --git a/test/test_persistence.py b/test/test_persistence.py index 7964543..6d123de 100644 --- a/test/test_persistence.py +++ b/test/test_persistence.py @@ -309,6 +309,10 @@ class TableTestCase(unittest.TestCase): assert len(list(cols)) == 4, 'column count mismatch' assert 'date' in cols and 'temperature' in cols and 'place' in cols + def test_drop_column(self): + self.tbl.drop_column('date') + assert 'date' not in self.tbl.columns + def test_iter(self): c = 0 for row in self.tbl: