def test_two_short_line_endpoints_get_picked(self): lines = [LineSegment(Point(0, 0), Point(1, 0)), \ LineSegment(Point(0, 1), Point(1, 1))] traj_lines = [TrajectoryLineSegment(lines[0], 0), \ TrajectoryLineSegment(lines[1], 1)] res = get_representative_trajectory_average_inputs(trajectory_line_segments=traj_lines, \ min_prev_dist=1, min_lines=2) expected = 2 self.assertEquals(len(res), expected)
def test_four_line_sets_should_result(self): lines = [LineSegment(Point(0, 0), Point(1, 0)), \ LineSegment(Point(0.5, 1), Point(1.5, 1))] traj_lines = [TrajectoryLineSegment(lines[0], 0), \ TrajectoryLineSegment(lines[1], 1)] res = get_representative_trajectory_average_inputs(trajectory_line_segments=traj_lines, \ min_prev_dist=0.5, min_lines=1) expected = 4 self.assertEquals(len(res), expected)
def verify(self, test_ob): self.assertEquals(1, 1) trajectory_average_inputs = \ get_representative_trajectory_average_inputs(test_ob['trajectory_line_segments'], test_ob['min_lines'], \ test_ob['min_prev_dist']) expected_line_sets = test_ob['expected_line_sets'] self.assertEquals(len(expected_line_sets), len(trajectory_average_inputs)) for i in xrange(0, len(trajectory_average_inputs)): averaging_input = trajectory_average_inputs[i] self.assertEquals(len(averaging_input['lines']), len(expected_line_sets[i]['lines']), \ "expected: " + str(len(expected_line_sets[i]['lines'])) + \ " but found: " + str(len(averaging_input['lines'])) + \ ". comparison failed for line set number " + str(i)) for line in averaging_input['lines']: self.assertTrue(line in expected_line_sets[i]['lines']) self.assertAlmostEquals(averaging_input['horizontal_position'], \ expected_line_sets[i]['horizontal_position'], 11, DECIMAL_MAX_DIFF_FOR_EQUALITY)