def test_to_instances(self):
t = TypedDfBuilder("T").require("animal", dtype=str).reserve("age", dtype=int).build()
df: t = t.of(
[
pd.Series(dict(animal="goldfish", age=2)),
pd.Series(dict(animal="goldfish", age=1)),
pd.Series(dict(animal="gazelle", age=8)),
pd.Series(dict(animal="pineapple", age=114)),
pd.Series(dict(animal="anteater", age=11)),
]
)
dc = t.create_dataclass()
instances = df.to_dataclass_instances()
assert instances == [
dc("goldfish", 2),
dc("goldfish", 1),
dc("gazelle", 8),
dc("pineapple", 114),
dc("anteater", 11),
]
assert list(sorted(instances)) == [
dc("anteater", 11),
dc("gazelle", 8),
dc("goldfish", 1),
dc("goldfish", 2),
dc("pineapple", 114),
]
def test_to_instances_empty(self):
t = TypedDfBuilder("T").reserve("animal", dtype=str).build()
df: t = t.of([])
instances = df.to_dataclass_instances()
assert instances == []
