def testSmoothDeviationOneZigZag(self): tp = json.load(open("emission/tests/data/smoothing_data/caltrain_one_zigzag.json")) (retained_indices, removed_indices) = tags.smooth_deviation(tp) logging.info("removed indices = %s" % removed_indices) self.assertEqual(len(removed_indices), 5) self.assertEqual(removed_indices, [37, 38, 45, 46, 68])
def testSmoothDeviationMultiZigZag(self): tp = json.load(open("emission/tests/data/smoothing_data/caltrain_multi_zigzag.json")) (retained_indices, removed_indices) = tags.smooth_deviation(tp) logging.info("removed indices = %s" % removed_indices) self.assertEqual(len(removed_indices), 3) self.assertEqual(removed_indices, [21, 22, 57])
def testSmoothDeviationWalkOneZigZag(self): tp = json.load(open("emission/tests/data/smoothing_data/walk_one_zigzag.json")) (retained_indices, removed_indices) = tags.smooth_deviation(tp) logging.info("removed indices = %s" % removed_indices) self.assertEqual(len(removed_indices), 4) self.assertEqual(removed_indices, [5, 6, 7, 8])