def test_map_field_default_value(self):
    """Tests the default value for the `fields` argument."""
    graph = graphs.GraphsTuple(**self.graph)
    mapped_fields = []
    graph = graph.map(mapped_fields.append)
    self.assertCountEqual(
        mapped_fields, [graphs.EDGES, graphs.GLOBALS, graphs.NODES])
Tests the default value for the `fields` argument.
graph_nets/tests/graphs_test.py