def test_example_crossings_were_computed_correctly(self):
thresholds = session.query(models.Threshold).filter_by(determination=DETERMINATION).all()
for threshold in thresholds:
for cg in threshold.crossing_groups:
crossings = list(cg.crossings)
for _ in range(N_CROSSINGS_TO_CHECK_PER_CROSSING_GROUP):
traj = np.random.choice(crossings).trajectory
# recompute this trajectory's crossings
segments, peaks = time_series.segment_by_threshold(traj.odors(session),
threshold.value,
traj.timepoint_ids_extended)
traj_crossings = session.query(models.Crossing).\
filter_by(trajectory=traj, crossing_group=cg).\
order_by(models.Crossing.crossing_number)
for ctr, (segment, peak) in enumerate(zip(segments, peaks)):
self.assertEqual(segment[0], traj_crossings[ctr].start_timepoint_id)
self.assertEqual(segment[1], traj_crossings[ctr].entry_timepoint_id)
self.assertEqual(segment[2], traj_crossings[ctr].peak_timepoint_id)
self.assertEqual(segment[3], traj_crossings[ctr].exit_timepoint_id + 1)
self.assertEqual(segment[4], traj_crossings[ctr].end_timepoint_id + 1)
self.assertAlmostEqual(peak, traj_crossings[ctr].max_odor)
def test_crossing_groups_were_stored_correctly(self):
for expt in self.expts:
# make sure each experiment has a threshold with the correct determination
# and that each threshold has three crossing groups, one for each odor state
# and that at least one crossing exists per crossing group
threshold = session.query(models.Threshold).\
filter_by(experiment=expt, determination=DETERMINATION).first()
self.assertTrue(threshold is not None)
self.assertEqual(len(threshold.crossing_groups), 3)
odor_states = [cg.odor_state for cg in threshold.crossing_groups]
self.assertEqual(set(odor_states), set(ODOR_STATES))
for cg in threshold.crossing_groups:
self.assertGreater(len(cg.crossings), 0)
def setUp(self):
print("In test '{}'...".format(self._testMethodName))
self.expts = session.query(models.Experiment)
