def test__build_events_df_empty(): events = np.array([], dtype=np.float64) returned = analysis._build_events_df(events) assert returned.empty assert list(returned.columns) == ['start', 'end', 'score']
def test__build_events_df_events(): events = np.array([ [1.22297040e+09, 1.22299200e+09, 5.72643599e-01], [1.22301360e+09, 1.22303520e+09, 5.72643599e-01], [1.22351040e+09, 1.22353200e+09, 5.72643599e-01], ]) returned = analysis._build_events_df(events) expected = pd.DataFrame({ 'start': [1222970400, 1223013600, 1223510400], 'end': [1222992000, 1223035200, 1223532000], 'score': [0.572644, 0.572644, 0.572644] }) pd.testing.assert_frame_equal(returned, expected)