def test_split_by_observation_gap(self):
traj = make_traj([Node(), Node(minute=1), Node(minute=5), Node(minute=6)])
split = ObservationGapSplitter(traj).split(gap=timedelta(seconds=120))
assert type(split) == TrajectoryCollection
assert len(split) == 2
assert split.trajectories[0] == make_traj([Node(), Node(minute=1)], id='1_0')
assert split.trajectories[1] == make_traj([Node(minute=5), Node(minute=6)], id='1_1')
def test_splitbyobservationgap_does_not_alter_df(self):
traj = make_traj(
[Node(), Node(minute=1),
Node(minute=5),
Node(minute=7)])
traj_copy = traj.copy()
split = ObservationGapSplitter(traj).split(gap=timedelta(minutes=5))
assert_frame_equal(traj.df, traj_copy.df)
def test_split_by_observation_gap_skip_single_points(self):
traj = make_traj(
[Node(), Node(minute=1),
Node(minute=5),
Node(minute=7)])
split = ObservationGapSplitter(traj).split(gap=timedelta(seconds=61))
assert type(split) == TrajectoryCollection
assert len(split) == 1
assert split.trajectories[0] == make_traj(
[Node(), Node(minute=1)], id="1_0")
def test_collection_split_by_observation_gap(self):
split = ObservationGapSplitter(
self.collection).split(gap=timedelta(hours=1))
assert type(split) == TrajectoryCollection
assert len(split) == 4