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 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 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 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)
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)
