def test_ivl_to_dataframe(self): clnd = tb_12_days() clnd.add_schedule('my_schedule', lambda x: True) ivl = Interval(clnd, (2, 8)) clnd_df = clnd.to_dataframe(2, 8) ivl_df = ivl.to_dataframe() assert len(ivl_df) == len(ivl) == len(clnd_df) assert list(ivl_df.columns) == list(clnd_df.columns) assert 'my_schedule' in list(ivl_df.columns)
def test_void_ivl_to_dataframe(self): clnd = tb_12_days() clnd.add_schedule('my_schedule', lambda x: True) ivl = Interval(clnd, (2, 8)) void_ivl = _VoidInterval(clnd, (8, 2)) ivl_df_columns = list(ivl.to_dataframe().columns) void_ivl_df = void_ivl.to_dataframe() assert void_ivl_df.empty assert list(void_ivl_df.columns) == ivl_df_columns assert 'my_schedule' in list(void_ivl_df.columns)