def test_remove_segments():
    """
    """

    trajs = data.brownian_trajs_df()
    trajs = Trajectories(trajs)

    trajs.remove_segments(1, inplace=True)

    assert np.all(trajs.labels == [0, 2, 3, 4])