def test_map_field_is_parallel(self):
    """Tests that fields are mapped parallelaly, not sequentially."""
    graph = graphs.GraphsTuple(**self.graph)
    graph = graph.map(lambda v: None, ["edges", "receivers", "senders"])
Tests that fields are mapped parallelaly, not sequentially.
graph_nets/tests/graphs_test.py