def test_table_update():
result = list(
Table.from_rows([(1, ), (2, )],
["a"]).update(b=c.col("a") +
1).into_iter_rows(include_header=True))
assert result == [
("a", "b"),
(1, 2),
(2, 3),
]
result = list(
Table.from_rows(result,
True).update(c=c.col("a") + c.col("b") + 1).update(
d=c.col("c") * -1).take("a", "c",
"d").into_iter_rows(dict))
assert result == [
{
"a": 1,
"c": 4,
"d": -4
},
{
"a": 2,
"c": 6,
"d": -6
},
]
def test_table_filter():
result = list(
Table.from_rows(
[(-1, 0), (1, 2),
(3, 4)], ["a", "b"]).filter(c.col("a") > 0).into_iter_rows(dict))
assert result == [{"a": 1, "b": 2}, {"a": 3, "b": 4}]
result = list(
Table.from_rows([(-1, 0), (1, 2), (3, 4)],
["a", "b"]).update(c=c.col("a") * 100).filter(
c.col("c") > 0).drop("c").into_iter_rows(dict))
assert result == [{"a": 1, "b": 2}, {"a": 3, "b": 4}]
def test_table_left_simple():
result = list(
Table.from_rows([(1, 2), (2, 3)], ["a", "b"]).join(
Table.from_rows([(1, 3), (2, 4)], ["a", "c"]),
how="left",
on=["a"],
).into_iter_rows(include_header=True))
assert result == [
("a", "b", "c"),
(1, 2, 3),
(2, 3, 4),
]
def test_table_right_join():
result = list(
Table.from_rows([(1, 2), (2, 3)], ["a", "b"]).join(
Table.from_rows([(4, 3), (2, 4)], ["a", "c"]),
how="right",
on=["a"],
).into_iter_rows(include_header=True))
assert result == [
("a", "b", "c"),
(4, None, 3),
(2, 3, 4),
]
def test_table_outer_join():
result = list(
Table.from_rows([(1, 2, 10), (2, 3, 11)], ["a", "b", "d"]).join(
Table.from_rows([(4, 3, 20), (2, 4, 21)], ["a", "c", "d"]),
how="outer",
on=["a"],
).into_iter_rows(include_header=True))
assert result == [
("a", "b", "d_LEFT", "c", "d_RIGHT"),
(1, 2, 10, None, None),
(2, 3, 11, 4, 21),
(4, None, None, 3, 20),
]
def test_table_exceptions():
with pytest.raises(c.ConversionException):
c.col("tst").gen_converter()
with pytest.raises(ValueError):
c.col(1)
with pytest.raises(ValueError):
ColumnDef("abc", None, None)
with pytest.raises(ValueError):
ColumnDef("abc", 0, c.this)
with pytest.raises(ValueError):
MetaColumns(duplicate_columns="abc")
with pytest.raises(ValueError):
Table.from_rows([("a", ), (1, )], True).take("b")
with pytest.raises(ValueError):
Table.from_rows([("a", ), (1, )], True).drop("b")
with pytest.raises(ValueError):
it = (Table.from_rows([(1, 2), (2, 3)], ["a", "b"]).join(
Table.from_rows([(0, -1), (3, 4)], ["a", "c"]),
how="inner",
on=c.LEFT.col("a") < c.col("d"),
).into_iter_rows(include_header=True))
table = Table.from_rows([(1, 2), (2, 3)], ["a", "b"])
with pytest.raises(TypeError):
table.into_iter_rows(object)
with pytest.raises(ValueError):
table.rename(["A"])
with pytest.raises(TypeError):
table.rename({"A"})
for header in [("a", ), ["a"], {"a": "a"}]:
with pytest.raises(ValueError):
table.from_rows([[1, 2], [2, 3]], header=header)
def test_table_drop():
result = list(
Table.from_rows([(1, 2, 3), (2, 3, 4)], ["a", "b", "c"]).drop(
"a", "c").into_iter_rows(include_header=True))
assert result == [
("b", ),
(2, ),
(3, ),
]
def test_table_rename():
result = list(
Table.from_rows([(1, 2), (3, 4)],
["a", "b"]).rename(["A", "B"]).into_iter_rows(dict))
assert result == [{"A": 1, "B": 2}, {"A": 3, "B": 4}]
result = list(
Table.from_rows([(1, 2), (3, 4)], ["a", "b"]).rename({
"b": "B"
}).into_iter_rows(tuple, include_header=True))
assert result == [("a", "B"), (1, 2), (3, 4)]
result = list(
Table.from_rows([{
"a": 1
}]).rename({
"a": "A"
}).into_iter_rows(dict))
assert result == [{"A": 1}]
def test_table_take():
result = list(
Table.from_rows([(1, 2, 3), (2, 3, 4)], ["a", "b", "c"],
duplicate_columns="keep").take(
"c", "a", "a").into_iter_rows(include_header=True))
assert result == [
("c", "a", "a"),
(3, 1, 1),
(4, 2, 2),
]
def test_table_left_join():
with c.OptionsCtx() as options:
options.debug = False
result = list(
Table.from_rows([(1, 2), (2, 3)], ["a", "b"]).join(
Table.from_rows([(1, 3), (2, 4)], ["a", "c"]),
how="left",
on=["a"],
).join(
Table.from_rows(
[
(2, 4, 7),
],
["a", "c", "d"],
),
how="left",
on=["a", "c"],
).into_iter_rows(include_header=True))
assert result == [
("a", "b", "c", "d"),
(1, 2, 3, None),
(2, 3, 4, 7),
]
def test_table_zip():
result = list(
Table.from_rows([["a", "b"], [1, 2]], header=True).zip(
Table.from_rows([["b", "a"], [4, 3], [6, 5]], header=True),
fill_value=False,
).into_iter_rows(tuple, include_header=True))
assert result == [
("a", "b", "b", "a"),
(1, 2, 4, 3),
(False, False, 6, 5),
]
result = list(
Table.from_rows([["a"], ["1"], ["2"], ["3"]], header=True).update_all(
int, int).zip(
Table.from_rows([["a"], ["4"]], header=True).update(
a=c.col("a").as_type(int))).into_iter_rows(
tuple, include_header=True))
assert result == [
("a", "a"),
(1, 4),
(2, None),
(3, None),
]
def test_table_inner_join():
result = list(
Table.from_rows([(1, 2), (2, 3)], ["a", "b"]).join(
Table.from_rows([(1, 3), (2, 4)], ["a", "c"]),
how="inner",
on=["a"],
).into_iter_rows(dict))
assert result == [
{
"a": 1,
"b": 2,
"c": 3
},
{
"a": 2,
"b": 3,
"c": 4
},
]
result = list(
Table.from_rows([(1, 2), (2, 3)],
["a", "b"]).embed_conversions().update().drop().join(
Table.from_rows([(1, 3), (2, 4)], ["a", "c"]),
how="inner",
on=["a"],
).into_iter_rows(include_header=True))
assert result == [("a", "b", "c"), (1, 2, 3), (2, 3, 4)]
with c.OptionsCtx() as o:
o.debug = True
result = list(
Table.from_rows([(1, 2), (2, 3)], ["a", "b"]).join(
Table.from_rows([(0, -1), (3, 4)], ["a", "c"]),
how="inner",
on=c.LEFT.col("a") < c.RIGHT.col("a"),
).into_iter_rows(include_header=True))
assert result == [
("a_LEFT", "b", "a_RIGHT", "c"),
(1, 2, 3, 4),
(2, 3, 3, 4),
]
def test_table_base_init():
list(
Table.from_rows([(1, -2), (2, -3)], ["a", "b"]).update(
c=c.col("a") + c.col("b")) # adding new column: "c"
.update(c=c.call_func(abs, c.col("c"))) # updating new column: "c"
.into_iter_rows(dict))
result = list(
Table.from_rows([(1, 2, 3), (2, 3, 4)],
["a", "b", "c"]).into_iter_rows(include_header=True))
assert result == [
("a", "b", "c"),
(1, 2, 3),
(2, 3, 4),
]
result = list(
Table.from_rows([(1, 2, 3), (2, 3, 4)], {
"a": 2,
"b": 1,
"c": 0
}).into_iter_rows(dict))
assert result == [
{
"a": 3,
"b": 2,
"c": 1
},
{
"a": 4,
"b": 3,
"c": 2
},
]
input_data = [("a", "a", "b"), (1, 2, 3)]
with pytest.raises(ValueError):
Table.from_rows(input_data, True)
with pytest.raises(ValueError):
Table.from_rows(input_data, True, duplicate_columns="raise")
result = list(
Table.from_rows(
input_data, True,
duplicate_columns="keep").into_iter_rows(include_header=True))
assert result == input_data
result = list(
Table.from_rows(
input_data, True,
duplicate_columns="drop").into_iter_rows(include_header=True))
assert result == [("a", "b"), (1, 3)]
result = list(
Table.from_rows(
input_data, True,
duplicate_columns="mangle").into_iter_rows(include_header=True))
assert result == [("a", "a_1", "b"), (1, 2, 3)]
result = list(
Table.from_rows(input_data, None).into_iter_rows(include_header=True))
assert result == [
("COLUMN_0", "COLUMN_1", "COLUMN_2"),
("a", "a", "b"),
(1, 2, 3),
]
result = list(
Table.from_rows(input_data, {
"a": 0,
"b": 1,
"c": 2
}).into_iter_rows(include_header=True))
assert result == [("a", "b", "c"), ("a", "a", "b"), (1, 2, 3)]
result = list(
Table.from_rows(input_data, {
"a": 1,
"b": 0,
"c": 2
}, skip_rows=1).into_iter_rows(dict))
assert result == [{"a": 2, "b": 1, "c": 3}]
result = list(
Table.from_rows([{
"a": 1,
"b": 2,
"c": 3
}]).into_iter_rows(dict))
assert result == [{"a": 1, "b": 2, "c": 3}]
result = list(
Table.from_rows([{
"a": 1,
"b": 2,
"c": 3
}], header=False).into_iter_rows(dict))
assert result == [{"COLUMN_0": 1, "COLUMN_1": 2, "COLUMN_2": 3}]
assert list(
Table.from_rows([1, (1, ), (2, )],
header=True).update(**{
"abc": c.col("1").item(0)
}).take("abc").into_iter_rows(dict)) == [
{
"abc": 1
},
{
"abc": 2
},
]
Table.from_rows(range(3), header=False).update(a=c.col("COLUMN_0"))
assert list(
Table.from_rows(["name", "cde"],
header=True).into_iter_rows(dict)) == [{
"name": "cde"
}]
assert list(
Table.from_rows(["name", "cde"],
header=False).into_iter_rows(dict)) == [{
"COLUMN_0":
"name"
}, {
"COLUMN_0":
"cde"
}]
def test_table_chain():
with c.OptionsCtx() as o:
o.debug = True
result = list(
Table.from_rows([["a", "b"], [1, 2]], header=True).chain(
Table.from_rows([["b", "a", "c"], [4, 3, 5]],
header=True)).into_iter_rows(
tuple, include_header=True))
assert result == [
("a", "b", "c"),
(1, 2, None),
(3, 4, 5),
]
result = list(
Table.from_rows([["a", "b"], [1, 2], [5, 6]], header=True).chain(
Table.from_rows([["c", "a"], [4, 3]],
header=True)).into_iter_rows(tuple,
include_header=True))
assert result == [
("a", "b", "c"),
(1, 2, None),
(5, 6, None),
(3, None, 4),
]
result = list(
Table.from_rows(
[["a"], ["1"]],
header=True).update(a=c.col("a").as_type(int) + 10).chain(
Table.from_rows([["b"], ["2"]], header=True).update_all(int),
fill_value=False,
).into_iter_rows(tuple, include_header=True))
assert result == [
("a", "b"),
(11, False),
(False, 2),
]
result = list(
Table.from_rows([["a"], ["1"]], header=True).chain(
Table.from_rows([["a"], ["2"]],
header=True)).into_iter_rows(include_header=True))
assert result == [
("a", ),
("1", ),
("2", ),
]
result = list(
Table.from_rows([["a"], ["1"]],
header=True).update(a=c.col("a").as_type(int)).chain(
Table.from_rows([["a"], ["2"]],
header=True)).into_iter_rows())
assert result == [(1, ), ("2", )]
result = list(
Table.from_rows([["a"], ["1"]], header=True).chain(
Table.from_rows([["a"], ["2"]], header=True).update(
a=c.col("a").as_type(int))).into_iter_rows())
assert result == [("1", ), (2, )]
result = list(
Table.from_rows([["a"], ["1"]], header=True).update(
a=c.col("a").as_type(int)).embed_conversions().chain(
Table.from_rows([["a"], ["2"]], header=True)).into_iter_rows())
assert result == [(1, ), ("2", )]
def test_table_integration():
input_data = [["a", "b"], [1, 2], [3, 4]]
conversion = c.this.pipe(lambda it: Table.from_rows(it, header=True).
into_iter_rows(dict)).as_type(list)
conversion.execute(input_data)
def process_table(rows):
return (Table.from_rows(
rows, ["a", "b"]).filter(c.col("a") > 0).into_iter_rows(dict))
def test_table_csv():
Table.from_csv("tests/csvs/ab.csv", True).into_csv("tests/csvs/out.csv")
result = list(Table.from_csv("tests/csvs/out.csv", True).into_iter_rows())
assert result == [("1", "2"), ("2", "3")]
result = list(
Table.from_csv("tests/csvs/ab.csv", True).update_all(int).join(
Table.from_csv(
"tests/csvs/ac.csv",
True,
dialect=Table.csv_dialect(delimiter="\t"),
).update_all(int),
on=["a"],
how="inner",
).into_iter_rows(include_header=True))
assert result == [
("a", "b", "c"),
(1, 2, 3),
(2, 3, 4),
]
with open("tests/csvs/ab.csv", "r") as f_in, open("tests/csvs/out.csv",
"w") as f_out:
result = Table.from_csv(f_in, header=["A", "B"],
skip_rows=True).into_csv(f_out)
result = list(Table.from_csv("tests/csvs/out.csv").into_iter_rows())
assert result == [("A", "B"), ("1", "2"), ("2", "3")]
Table.from_rows([{"a": 1}]).into_csv("tests/csvs/out.csv")
result = list(Table.from_csv("tests/csvs/out.csv").into_iter_rows())
assert result == [("a", ), ("1", )]
Table.from_rows([{
"a": 1
}]).into_csv("tests/csvs/out.csv", include_header=False)
result = list(Table.from_csv("tests/csvs/out.csv").into_iter_rows())
assert result == [("1", )]
with open("tests/csvs/out.csv", "w") as f_out:
Table.from_rows([{"a": 1}]).into_csv(f_out, include_header=False)
result = list(Table.from_csv("tests/csvs/out.csv").into_iter_rows())
assert result == [("1", )]
def test_table_columns():
ref_columns = ["a", "b"]
table = Table.from_rows([(1, 2), (3, 4)], ref_columns)
columns = table.get_columns()
columns2 = table.columns
assert columns == ref_columns and columns2 == ref_columns
