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()