def test_add_traj_id_can_overwrite(self):
df = self.default_traj_metric.df.copy()
df[TRAJ_ID_COL_NAME] = 1
traj = Trajectory(df, "b")
traj.add_traj_id(overwrite=True)
assert traj.df[TRAJ_ID_COL_NAME].tolist() == ["b", "b", "b"]
def test_add_traj_id_overwrite_raises_error(self):
df = self.default_traj_metric.df.copy()
df[TRAJ_ID_COL_NAME] = 1
traj = Trajectory(df, "b")
with pytest.raises(RuntimeError):
traj.add_traj_id()
