def test_line_segment_generator(self): lines = [self.create_trajectory_line_seg((0, 0), (1, 2), 0, 3), \ self.create_trajectory_line_seg((3, 2), (3, 4), 1, 2), \ self.create_trajectory_line_seg((4, 5), (2, 3), 2, 4)] test_ob = {'horizontal_position':3, 'lines': [] } for line in lines: test_ob['lines'].append(line) expected = map(lambda seg: {'horizontal_pos':3, 'line_seg': seg.line_segment }, lines) self.verify_iterable_works_more_than_once(line_segment_averaging_set_iterable(test_ob), expected)
def test_empty_lines_for_line_segment_generator(self): test_ob = {'horizontal_position': 4, 'lines': []} self.verify_iterable_works_more_than_once(line_segment_averaging_set_iterable(test_ob), [])