def test_grab_machine_coordinates(self, mck_paths): """ Retrieve machine coordinates for Point, LineString and Polygon format. """ mck_paths.COORDINATES = self.coords_path coords_b = coordinates.component_coordinates("machine-B") coords_c = coordinates.component_coordinates("machine-C") coords_d = coordinates.component_coordinates("machine-D") self.assertEqual(coords_b, { "type": "Point", "coordinates": [-78.254, 40.45712] }) self.assertEqual( coords_c, { "type": "LineString", "coordinates": [[102.0, 0.0], [103.0, 1.0], [104.0, 0.0], [105.0, 1.0]] }) self.assertEqual( coords_d, { "type": "Polygon", "coordinates": [[[100.0, 0.0], [101.0, 0.0], [101.0, 1.0], [100.0, 1.0], [100.0, 0.0]]] })
def test_none_name(self, mck_paths): """ Check that a None value input is ignored. """ mck_paths.COORDINATES = self.coords_path coords = coordinates.component_coordinates(None) self.assertIsNone(coords)
def test_unknown_name(self, mck_paths): """ Check that a name that does not exist is ignored. """ mck_paths.COORDINATES = self.coords_path coords = coordinates.component_coordinates("machine-G") self.assertIsNone(coords)
def _add_coordinates(graph, node): """ Adds coordinates to the node within the graph. As the graph is passed by reference the calling method gets back the graph with the coordinates added. :param graph: The graph containing the node. :param node: THe id of the node. """ coords = coordinates.component_coordinates(node) graph.node[node]["attributes"]["geo"] = coords