def test_dataframes_read_yaml(self):
values = [
V1Event(
timestamp=dt_parser.parse("2018-12-11 10:24:57"),
dataframe=V1EventDataframe(path="path1",
content_type="parquet"),
step=12,
),
V1Event(
timestamp=dt_parser.parse("2018-12-11 10:25:57"),
dataframe=V1EventDataframe(path="path2",
content_type="pickle"),
step=13,
),
V1Event(
timestamp=dt_parser.parse("2018-12-11 10:26:57"),
dataframe=V1EventDataframe(path="path3"),
step=14,
),
]
events = V1Events.read(
name="foo",
kind="dataframe",
data=os.path.abspath(
"tests/fixtures/polyboard/dataframe/dataframe_events.plx"),
)
assert events.name == "foo"
assert len(events.df.values) == 3
for i in range(3):
assert events.get_event_at(i).to_dict() == values[i].to_dict()
def test_dataframe(self):
events = LoggedEventListSpec(
name="foo",
kind="dataframe",
events=[
V1Event(
timestamp=dt_parser.parse("2018-12-11 10:24:57"),
dataframe=V1EventDataframe(path="path",
content_type="parquet"),
step=12,
),
V1Event(
timestamp=dt_parser.parse("2018-12-11 11:24:57"),
dataframe=V1EventDataframe(path="path",
content_type="pickle"),
step=13,
),
V1Event(
timestamp=dt_parser.parse("2018-12-11 12:24:57"),
dataframe=V1EventDataframe(path="path"),
step=14,
),
],
)
events_dict = events.to_dict()
assert events_dict == events.from_dict(events_dict).to_dict()