def test_returns_none_none_when_no_possible_connections(self):
max_distance_for_neighbors_in_different_trajectories = 1.0
other_neighbors_func = get_find_other_nearby_neighbors_func(
max_distance_for_neighbors_in_different_trajectories)
pt_graph = build_point_graph(
[], add_other_neigbors_func=other_neighbors_func)
def dummy_find_other_neighbors_func(pt_node, pt_graph):
return []
compute_graph_component_ids(pt_graph=pt_graph, \
find_other_neighbors_func=dummy_find_other_neighbors_func)
start_pt = Point(1.5, 2.5)
end_pt = Point(4.5, 2.5)
max_dist_to_existing_pt = 0.9
actual_shortest_connection, actual_shortest_distance = find_shortest_connection(start_pt=start_pt, \
end_pt=end_pt, \
pt_graph=pt_graph, \
max_dist_to_existing_pt=max_dist_to_existing_pt)
self.assertEquals(actual_shortest_connection, None)
self.assertEquals(actual_shortest_distance, None)
def test_computes_components_correctly_when_all_in_same_component(self):
trajectories = [[], \
[Point(0, 0), Point(1, 1), Point(2, 2)], \
[], \
[Point(10, 0), Point(11, 0), Point(12, 0)], \
[Point(0, 10), Point(0, 11), Point(0, 12)], \
[]]
filtered_trajs = []
cur_index = 0
for traj in trajectories:
if len(traj) > 0:
filtered_trajs.append(FilteredTrajectory(traj, cur_index))
cur_index += 1
pt_graph = build_point_graph(filtered_trajectories=filtered_trajs)
def dummy_find_other_neighbors_func(pt_node, pt_graph):
other_neighbors_table = [[3], [], [], [0], [], [6], [5], [], []]
return other_neighbors_table[pt_node.index]
compute_graph_component_ids(pt_graph=pt_graph, \
find_other_neighbors_func=dummy_find_other_neighbors_func)
expected_component_ids = [0] * 9
for pt_node in pt_graph:
self.assertEquals(pt_node.graph_component_id, \
expected_component_ids[pt_node.index], \
"pt node comp id at index " + str(pt_node.index) + \
" is " + str(pt_node.graph_component_id) + \
" but expected " + \
str(expected_component_ids[pt_node.index]))
def test_computes_components_correctly_when_all_in_same_component(self):
trajectories = [[], \
[Point(0, 0), Point(1, 1), Point(2, 2)], \
[], \
[Point(10, 0), Point(11, 0), Point(12, 0)], \
[Point(0, 10), Point(0, 11), Point(0, 12)], \
[]]
filtered_trajs = []
cur_index = 0
for traj in trajectories:
if len(traj) > 0:
filtered_trajs.append(FilteredTrajectory(traj, cur_index))
cur_index += 1
pt_graph = build_point_graph(filtered_trajectories=filtered_trajs)
def dummy_find_other_neighbors_func(pt_node, pt_graph):
other_neighbors_table = [[3], [], [], [0], [], [6], [5], [], []]
return other_neighbors_table[pt_node.index]
compute_graph_component_ids(pt_graph=pt_graph, \
find_other_neighbors_func=dummy_find_other_neighbors_func)
expected_component_ids = [0] * 9
for pt_node in pt_graph:
self.assertEquals(pt_node.graph_component_id, \
expected_component_ids[pt_node.index], \
"pt node comp id at index " + str(pt_node.index) + \
" is " + str(pt_node.graph_component_id) + \
" but expected " + \
str(expected_component_ids[pt_node.index]))
def construct_graph_from_processed_trajectories(filtered_trajectories, \
max_distance_between_connected_different_trajs):
other_neighbors_func = \
get_find_other_nearby_neighbors_func(max_distance_between_connected_different_trajs)
cur_index = 0
graph_input = []
for traj in filtered_trajectories:
point_traj = map(lambda x: Point(x['lat'], x['lng']), traj)
graph_input.append(FilteredTrajectory(point_traj, cur_index))
cur_index += 1
def dummy_find_other_neighbors_func(pt_node, pt_graph):
return []
pt_graph = build_point_graph(graph_input, other_neighbors_func)
compute_graph_component_ids(pt_graph=pt_graph, \
find_other_neighbors_func=dummy_find_other_neighbors_func)
return pt_graph
def test_returns_none_none_when_no_possible_connections(self):
max_distance_for_neighbors_in_different_trajectories = 1.0
other_neighbors_func = get_find_other_nearby_neighbors_func(max_distance_for_neighbors_in_different_trajectories)
pt_graph = build_point_graph([], add_other_neigbors_func=other_neighbors_func)
def dummy_find_other_neighbors_func(pt_node, pt_graph):
return []
compute_graph_component_ids(pt_graph=pt_graph, \
find_other_neighbors_func=dummy_find_other_neighbors_func)
start_pt = Point(1.5, 2.5)
end_pt = Point(4.5, 2.5)
max_dist_to_existing_pt = 0.9
actual_shortest_connection, actual_shortest_distance = find_shortest_connection(start_pt=start_pt, \
end_pt=end_pt, \
pt_graph=pt_graph, \
max_dist_to_existing_pt=max_dist_to_existing_pt)
self.assertEquals(actual_shortest_connection, None)
self.assertEquals(actual_shortest_distance, None)
def test_computes_shortest_connection_correctly(self):
trajectories = [[Point(1, 2), Point(2, 2)], \
[Point(1, 3), Point(2, 3)], \
[Point(1, 4), Point(2, 4), Point(3, 4), Point(4, 4)], \
[Point(4, 3), Point(5, 3)], \
[Point(4, 2), Point(5, 2)], \
[Point(1, 5), Point(1, 6)], \
[Point(7, 4), Point(8, 4)], \
[Point(7, 5), Point(8, 5)]]
filtered = []
cur_index = 0
for traj in trajectories:
filtered.append(FilteredTrajectory(traj, cur_index))
cur_index += 1
max_distance_for_neighbors_in_different_trajectories = 1.0
other_neighbors_func = get_find_other_nearby_neighbors_func(
max_distance_for_neighbors_in_different_trajectories)
pt_graph = build_point_graph(
filtered, add_other_neigbors_func=other_neighbors_func)
expected_neighbor_indices = [[1, 2], \
[0, 3], \
[0, 3, 4], \
[1, 2, 5], \
[2, 5, 12], \
[3, 4, 6], \
[5, 7], \
[6, 8], \
[7, 9, 10], \
[8, 11], \
[8, 11], \
[9, 10], \
[4, 13], \
[12], \
[15, 16], \
[14, 17], \
[14, 17], \
[15, 16]]
for pt_node in pt_graph:
for i in expected_neighbor_indices[pt_node.index]:
self.assertTrue(i in pt_node.get_neighbor_indices(), "node index " + \
str(i) + " is not in node index " + str(pt_node.index) + \
" neighbor list")
find_other_neighbors_func = \
get_find_other_nearby_neighbors_func(max_distance_for_neighbors_in_different_trajectories)
def dummy_find_other_neighbors_func(pt_node, pt_graph):
return []
compute_graph_component_ids(pt_graph=pt_graph, \
find_other_neighbors_func=dummy_find_other_neighbors_func)
start_pt = Point(1.5, 2.5)
end_pt = Point(4.5, 2.5)
max_dist_to_existing_pt = 0.9
actual_shortest_connection, actual_shortest_distance = find_shortest_connection(start_pt=start_pt, \
end_pt=end_pt, \
pt_graph=pt_graph, \
max_dist_to_existing_pt=max_dist_to_existing_pt)
expected_shortest_connection = [Point(2, 3), \
Point(2, 4), \
Point(3, 4), \
Point(4, 4), \
Point(4, 3)]
# Note that the shortest path distance and calculation ignores
# the initial distance to a point on the map
expected_shortest_distance = 4.0
self.assertEquals(len(expected_shortest_connection),
len(actual_shortest_connection))
exp_iter = iter(expected_shortest_connection)
actual_iter = iter(actual_shortest_connection)
for exp_pt in exp_iter:
act_pt = actual_iter.next()
self.assertEquals(exp_pt.x, act_pt.x)
self.assertEquals(exp_pt.y, act_pt.y)
self.assertEquals(expected_shortest_distance, actual_shortest_distance)
def test_computes_shortest_connection_correctly(self):
trajectories = [[Point(1, 2), Point(2, 2)], \
[Point(1, 3), Point(2, 3)], \
[Point(1, 4), Point(2, 4), Point(3, 4), Point(4, 4)], \
[Point(4, 3), Point(5, 3)], \
[Point(4, 2), Point(5, 2)], \
[Point(1, 5), Point(1, 6)], \
[Point(7, 4), Point(8, 4)], \
[Point(7, 5), Point(8, 5)]]
filtered = []
cur_index = 0
for traj in trajectories:
filtered.append(FilteredTrajectory(traj, cur_index))
cur_index += 1
max_distance_for_neighbors_in_different_trajectories = 1.0
other_neighbors_func = get_find_other_nearby_neighbors_func(max_distance_for_neighbors_in_different_trajectories)
pt_graph = build_point_graph(filtered, add_other_neigbors_func=other_neighbors_func)
expected_neighbor_indices = [[1, 2], \
[0, 3], \
[0, 3, 4], \
[1, 2, 5], \
[2, 5, 12], \
[3, 4, 6], \
[5, 7], \
[6, 8], \
[7, 9, 10], \
[8, 11], \
[8, 11], \
[9, 10], \
[4, 13], \
[12], \
[15, 16], \
[14, 17], \
[14, 17], \
[15, 16]]
for pt_node in pt_graph:
for i in expected_neighbor_indices[pt_node.index]:
self.assertTrue(i in pt_node.get_neighbor_indices(), "node index " + \
str(i) + " is not in node index " + str(pt_node.index) + \
" neighbor list")
find_other_neighbors_func = \
get_find_other_nearby_neighbors_func(max_distance_for_neighbors_in_different_trajectories)
def dummy_find_other_neighbors_func(pt_node, pt_graph):
return []
compute_graph_component_ids(pt_graph=pt_graph, \
find_other_neighbors_func=dummy_find_other_neighbors_func)
start_pt = Point(1.5, 2.5)
end_pt = Point(4.5, 2.5)
max_dist_to_existing_pt = 0.9
actual_shortest_connection, actual_shortest_distance = find_shortest_connection(start_pt=start_pt, \
end_pt=end_pt, \
pt_graph=pt_graph, \
max_dist_to_existing_pt=max_dist_to_existing_pt)
expected_shortest_connection = [Point(2, 3), \
Point(2, 4), \
Point(3, 4), \
Point(4, 4), \
Point(4, 3)]
# Note that the shortest path distance and calculation ignores
# the initial distance to a point on the map
expected_shortest_distance = 4.0
self.assertEquals(len(expected_shortest_connection), len(actual_shortest_connection))
exp_iter = iter(expected_shortest_connection)
actual_iter = iter(actual_shortest_connection)
for exp_pt in exp_iter:
act_pt = actual_iter.next()
self.assertEquals(exp_pt.x, act_pt.x)
self.assertEquals(exp_pt.y, act_pt.y)
self.assertEquals(expected_shortest_distance, actual_shortest_distance)
