def test_example_trajectories_were_cleaned_correctly(self):
trajs_raw_not_clean = session.query(models.Trajectory).filter_by(
raw=True, clean=False).all()
for traj in trajs_raw_not_clean[:N_EXAMPLE_TRAJS_TO_CLEAN]:
# get cleaning parameters
insect = traj.experiment.insect
cleaning_params_list = session.query(models.TrajectoryCleaningParameter.param,
models.TrajectoryCleaningParameter.value).\
filter_by(insect=insect).all()
cleaning_params = dict(cleaning_params_list)
# clean traj
clean_portions = clean_traj(traj, cleaning_params)
# check to make sure the correct number of clean trajectories were generated with the right ids
# and that they have the right timepoint indicators
for ctr, clean_portion in enumerate(clean_portions):
clean_id = traj.id + '_c{}'.format(ctr)
traj_cleaned = session.query(models.Trajectory).get(clean_id)
self.assertFalse(traj_cleaned is None)
self.assertEqual(traj_cleaned.start_timepoint_id,
clean_portion[0])
self.assertEqual(traj_cleaned.end_timepoint_id,
clean_portion[1])
# make sure that there is no additional trajectory that was created
self.assertTrue(session.query(models.Trajectory). \
get(traj.id + '_c{}'.format(len(clean_portions))) is None)
def test_example_trajectories_were_cleaned_correctly(self):
trajs_raw_not_clean = session.query(models.Trajectory).filter_by(raw=True, clean=False).all()
for traj in trajs_raw_not_clean[:N_EXAMPLE_TRAJS_TO_CLEAN]:
# get cleaning parameters
insect = traj.experiment.insect
cleaning_params_list = session.query(models.TrajectoryCleaningParameter.param,
models.TrajectoryCleaningParameter.value).\
filter_by(insect=insect).all()
cleaning_params = dict(cleaning_params_list)
# clean traj
clean_portions = clean_traj(traj, cleaning_params)
# check to make sure the correct number of clean trajectories were generated with the right ids
# and that they have the right timepoint indicators
for ctr, clean_portion in enumerate(clean_portions):
clean_id = traj.id + '_c{}'.format(ctr)
traj_cleaned = session.query(models.Trajectory).get(clean_id)
self.assertFalse(traj_cleaned is None)
self.assertEqual(traj_cleaned.start_timepoint_id, clean_portion[0])
self.assertEqual(traj_cleaned.end_timepoint_id, clean_portion[1])
# make sure that there is no additional trajectory that was created
self.assertTrue(session.query(models.Trajectory). \
get(traj.id + '_c{}'.format(len(clean_portions))) is None)
def test_some_but_not_all_raw_trajectories_are_clean(self):
all_trajs = session.query(models.Trajectory).all()
raw_not_clean = session.query(models.Trajectory).filter_by(raw=True, clean=False).all()
clean_not_raw = session.query(models.Trajectory).filter_by(raw=False, clean=True).all()
raw_and_clean = session.query(models.Trajectory).filter_by(raw=True, clean=True).all()
not_raw_not_clean = session.query(models.Trajectory).filter_by(raw=False, clean=False).all()
for traj_set in (all_trajs, raw_not_clean, clean_not_raw, raw_and_clean):
self.assertGreater(len(traj_set), 0)
self.assertEqual(len(not_raw_not_clean), 0)
self.assertEqual(len(all_trajs), len(raw_not_clean) + \
len(clean_not_raw) + \
len(raw_and_clean) + \
len(not_raw_not_clean))
def test_some_but_not_all_raw_trajectories_are_clean(self):
all_trajs = session.query(models.Trajectory).all()
raw_not_clean = session.query(models.Trajectory).filter_by(
raw=True, clean=False).all()
clean_not_raw = session.query(models.Trajectory).filter_by(
raw=False, clean=True).all()
raw_and_clean = session.query(models.Trajectory).filter_by(
raw=True, clean=True).all()
not_raw_not_clean = session.query(models.Trajectory).filter_by(
raw=False, clean=False).all()
for traj_set in (all_trajs, raw_not_clean, clean_not_raw,
raw_and_clean):
self.assertGreater(len(traj_set), 0)
self.assertEqual(len(not_raw_not_clean), 0)
self.assertEqual(len(all_trajs), len(raw_not_clean) + \
len(clean_not_raw) + \
len(raw_and_clean) + \
len(not_raw_not_clean))
