def testSmoothBoundaryMultiZigZag(self):
tp = json.load(open("emission/tests/data/smoothing_data/caltrain_multi_zigzag.json"))
(retained_indices, removed_indices) = tags.smooth_boundary(tp)
logging.info("removed indices = %s" % removed_indices)
self.assertEqual(len(removed_indices), 15)
self.assertEqual(removed_indices, [21, 26, 28, 29, 30, 40, 41, 42, 43, 46, 47, 48, 49, 50, 57])
def testSmoothBoundaryOneZigZag(self):
tp = json.load(open("emission/tests/data/smoothing_data/caltrain_one_zigzag.json"))
(retained_indices, removed_indices) = tags.smooth_boundary(tp)
logging.info("removed indices = %s" % removed_indices)
self.assertEqual(len(removed_indices), 1)
self.assertEqual(removed_indices, [68])