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()