def test_create_frames_interpolates_missing_frame(): trajectory = [(1, datetime.datetime(2017, 1, 1, 0, 0, 0), 100.0, 100.0, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 1, 0), 200.0, 200.0, 0, 0)] expected_frames = [ Frame(0, 100.0, 100.0), Frame(1, 150.0, 150.0), Frame(2, 200.0, 200.0) ] actual_frames = create_frames(trajectory) assert actual_frames == expected_frames
def test_create_frames_groups_sample_in_same_minute(): trajectory = [ (1, datetime.datetime(2017, 1, 1, 0, 0, 0), 100.0, 100.5, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 0, 30), 100.0, 100.5, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 0, 59), 100.0, 100.5, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 1, 0), 105.0, 105.5, 0, 0) ] expected_frames = [ Frame(0, 100.0, 100.5), Frame(1, 100.0, 100.5), Frame(2, 105.0, 105.5) ] actual_frames = create_frames(trajectory) assert actual_frames == expected_frames
def test_create_frames_performs_sed_to_select_samples_in_same_frame(): trajectory = [ (1, datetime.datetime(2017, 1, 1, 0, 1, 0), 100.0, 100.0, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 1, 30), 100.0, 100.5, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 1, 40), 200.0, 100.5, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 1, 59), 149.0, 104.5, 0, 0), (1, datetime.datetime(2017, 1, 1, 0, 2, 0), 150.0, 105.5, 0, 0) ] expected_frames = [ Frame(2, 100.0, 100.0), Frame(3, 200.0, 100.5), Frame(4, 150.0, 105.5) ] actual_frames = create_frames(trajectory) assert actual_frames == expected_frames
def test_create_frames_does_not_interpolate_last_missing_frames(): trajectory = [(1, datetime.datetime(2017, 1, 1, 0, 57, 0), 100.0, 100.0, 0, 0)] expected_frames = [Frame(114, 100.0, 100.0)] actual_frames = create_frames(trajectory) assert actual_frames == expected_frames