def test_single_line_segs(self): points = [2, 1, 4] expected_line_segs = [18] res = get_trajectory_line_segments_from_points_iterable(point_iterable=points, \ trajectory_line_segment_factory=self.MockTrajectoryLineSegFactory(), \ trajectory_id=5, \ trajectory_partitioning_func=self.mock_partition_func, \ line_seg_from_points_func=self.mock_line_seg_from_points_func) self.verify_iterable_equals_list(iterable_ob=res, list_ob=expected_line_segs)
def test_normal_case(self): points = [0, 1, 2, 3, 4, 5, 6] expected_line_segs = [0, 18, 44] res = get_trajectory_line_segments_from_points_iterable(point_iterable=points, \ trajectory_line_segment_factory=self.MockTrajectoryLineSegFactory(), \ trajectory_id=5, \ trajectory_partitioning_func=self.mock_partition_func, \ line_seg_from_points_func=self.mock_line_seg_from_points_func) self.verify_iterable_works_more_than_once(res, expected_line_segs)
def test_normal_case_odd_filter_function(self): list = [3, 0, 5, 2, 9, 4, 15] partitioning_func = self.create_mock_partitioning_func(keeper_func=lambda x: True if x % 2 == 1 else False) traj_id = 3 expected = [6, 12, 18] res = get_trajectory_line_segments_from_points_iterable(point_iterable=list, \ trajectory_line_segment_factory=self.MockTrajectoryLineSegFactory(), \ trajectory_id=traj_id, \ trajectory_partitioning_func=partitioning_func, \ line_seg_from_points_func=self.mock_line_segment_from_points_func) self.verify_iterable_works_more_than_once(res, expected)
def test_normal_case(self): list = [0, 1, 2, 3, 6, 5, 12] partitioning_func = self.create_mock_partitioning_func(keeper_func=lambda x: True if x % 2 == 0 else False) traj_id = 4 expected = [8, 16, 24] res = get_trajectory_line_segments_from_points_iterable(point_iterable=list, \ trajectory_line_segment_factory=self.MockTrajectoryLineSegFactory(), \ trajectory_id=traj_id, \ trajectory_partitioning_func=partitioning_func, \ line_seg_from_points_func=self.mock_line_segment_from_points_func) self.verify_iterable_works_more_than_once(res, expected)