def test_relabel():
    """
    """
    trajs = Trajectories(data.brownian_trajs_df())
    trajs.columns = ['x', 'y', 'z', 'new_label', 't']
    trajs.relabel(inplace=True)

    new_values = [[1.933058243735795, -14.581064591435775, 11.603556633147544, 0.0],
                  [-12.862215173899491, -2.8611502446443238, -2.2738941196781424, 0.0],
                  [9.100887851132633, 2.837252570763561, 2.875753940450461, 0.0],
                  [-9.253860446235523, 11.345550876585719, 22.118203258275745, 0.0]]

    assert trajs.iloc[:4].values.tolist() == new_values

    trajs = Trajectories(data.brownian_trajs_df())
    trajs.columns = ['x', 'y', 'z', 'new_label', 't']
    trajs = trajs.relabel(inplace=False)

    new_values = [[1.933058243735795, -14.581064591435775, 11.603556633147544, 0.0],
                  [-12.862215173899491, -2.8611502446443238, -2.2738941196781424, 0.0],
                  [9.100887851132633, 2.837252570763561, 2.875753940450461, 0.0],
                  [-9.253860446235523, 11.345550876585719, 22.118203258275745, 0.0]]

    assert trajs.iloc[:4].values.tolist() == new_values