def test_optional_column_non_empty(self): """an non-empty optional column shoud be displayed""" tbl = TextTable("%foo %bar") tbl.optional_cols = [ 'bar' ] tbl.append({'foo': 'zap', 'bar': 'zop' }) self.assertEqual(str(tbl), """FOO BAR\n--- ---\nzap zop""")
def test_optional_column_empty(self): """an empty optional column should not be displayed""" tbl = TextTable("%foo %bar") tbl.optional_cols = [ 'bar' ] tbl.append({'foo': 'zap', 'bar': None }) self.assertEqual(str(tbl), """FOO\n---\nzap""")