def test_drop():
td = TabularData(["col1", "col2", "col3"])
td.append(["foo", "bar", "baz"])
assert list(td) == [["foo", "bar", "baz"]]
td.drop("col2")
assert td.keys() == ["col1", "col3"]
assert list(td) == [["foo", "baz"]]
def test_protected():
td = TabularData(["col1", "col2", "col3", "other"])
td.append(["foo", "bar", "baz", "other_val"])
td.protect("col1", "col2")
td.drop("col1", "col2", "col3", "other")
assert td.keys() == ["col1", "col2"]
assert list(td) == [["foo", "bar"]]
td.unprotect("col2")
td.drop("col1", "col2")
assert td.keys() == ["col1"]
assert list(td) == [["foo"]]
