def test_all_nodes_inside(self):
nodes = NodeSet()
nodes.add(1, -10, -10)
nodes.add(2, -5, -5)
selected_nodes = select_nodes_in_rectangle(nodes, -10, 10, -10, 10)
self.assertTrue(
nodes.get_nodes().keys() == selected_nodes.get_nodes().keys())
def test_some_nodes_inside(self):
nodes = NodeSet()
nodes.add(1, -10, -10)
nodes.add(2, -5, -5)
selected_nodes = select_nodes_in_rectangle(nodes, -10, 10, -10, 10)
self.assertTrue(
any(x in selected_nodes.get_nodes().keys()
for x in nodes.get_nodes().keys()))
def test_rand_cases(self):
nodes = NodeSet()
nodes.add(1, random.randint(-90, 90), random.randint(-180, 180))
nodes.add(2, random.randint(-90, 90), random.randint(-180, 180))
for k, n in nodes.get_nodes().items():
self.assertTrue(-90 <= n.lat <= 90 and -180 <= n.lng <= 180)
def test_edge_bounds(self):
nodes = NodeSet()
nodes.add(1, -90, -180)
nodes.add(2, 90, 180)
for k, n in nodes.get_nodes().items():
self.assertTrue(-90 <= n.lat <= 90 and -180 <= n.lng <= 180)