def test_table_empty(capsys):
td = TabularData(["Col1", "Col2", "Col3"])
assert dict(td.items()) == {"Col1": [], "Col2": [], "Col3": []}
assert td.columns == [[], [], []]
assert td.keys() == ["Col1", "Col2", "Col3"]
assert list(td) == []
assert td.Col1 == []
assert td.Col2 == []
assert td.Col3 == []
assert td[1:] == []
with pytest.raises(IndexError):
_ = td[1]
assert len(td) == 0
assert td.shape == (3, 0)
assert td.to_csv() == """Col1,Col2,Col3\r\n"""
td.render()
assert capsys.readouterr() == ("", "")
td.render(rich_table=True)
assert capsys.readouterr() == ("", "")
td.render(markdown=True)
assert capsys.readouterr() == (
"| Col1 | Col2 | Col3 |\n|--------|--------|--------|\n\n",
"",
)
td.rename("Col1", "Col11")
assert td.keys() == ["Col11", "Col2", "Col3"]
td.project("Col3", "Col11")
assert td.keys() == ["Col3", "Col11"]
def test_row_from_dict():
td = TabularData(["col1", "col2"])
td.row_from_dict({"col3": "value3", "col4": "value4"})
assert td.keys() == ["col1", "col2", "col3", "col4"]
assert dict(td.items()) == {
"col1": [""],
"col2": [""],
"col3": ["value3"],
"col4": ["value4"],
}
td.row_from_dict({"col3": "value3", "col5": "value5", "col6": "value6"})
assert td.keys() == ["col1", "col2", "col3", "col4", "col5", "col6"]
assert dict(td.items()) == {
"col1": ["", ""],
"col2": ["", ""],
"col3": ["value3", "value3"],
"col4": ["value4", ""],
"col5": ["", "value5"],
"col6": ["", "value6"],
}
assert td.shape == (6, 2)
assert list(td) == [
["", "", "value3", "value4", "", ""],
["", "", "value3", "", "value5", "value6"],
]
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"]]
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_fill_value():
td = TabularData(["col-1", "col-2", "col-3"], fill_value="?")
td.append(["foo"])
assert list(td) == [["foo", "?", "?"]]
td.extend(
[
["bar"],
["foobar", "foobar2"],
["f", "fo", "foo", "foob", "fooba", "foobar"],
]
)
assert list(td) == [
["foo", "?", "?"],
["bar", "?", "?"],
["foobar", "foobar2", "?"],
["f", "fo", "foo"],
]
td.insert(1, ["lorem"])
assert td[1] == ["lorem", "?", "?"]
td[0] = ["lorem", "ipsum"]
assert td[0] == ["lorem", "ipsum", "?"]
td[1:2] = [["f", "fo"]]
assert td[1:2] == [["f", "fo", "?"]]
td.add_column("col-4")
assert td.keys() == ["col-1", "col-2", "col-3", "col-4"]
assert td[0][3] == "?"
def test_dict_like_interfaces():
td = TabularData(["col-1", "col-2"])
td.extend([["foo", "bar"], ["foobar", "foobar"]])
assert td.keys() == ["col-1", "col-2"]
assert dict(td.items()) == {
"col-1": ["foo", "foobar"],
"col-2": ["bar", "foobar"],
}
assert td.as_dict() == [
{"col-1": "foo", "col-2": "bar"},
{"col-1": "foobar", "col-2": "foobar"},
]
assert td.as_dict(["col-1"]) == [{"col-1": "foo"}, {"col-1": "foobar"}]
