docstr: Tests that the fields are mapped are as expected.
docstr_tok: ["Tests","that","the","fields","are","mapped","are","as","expected","."]
code: def test_map_fields_as_expected(self, fields_to_map):
      """Tests that the fields are mapped are as expected."""
      graph = graphs.GraphsTuple(**self.graph)
      graph = graph.map(lambda v: v + v, fields_to_map)
      for field in graphs.ALL_FIELDS:
          if field in fields_to_map:
              self.assertEqual(field + field, getattr(graph, field))
          else:
              self.assertEqual(field, getattr(graph, field))
  
file_path: graph_nets/graph_nets/tests/graphs_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  from graph_nets import graphs
  import tensorflow
label_desc: Use image node_id 6 for calling the GraphsTest obj's underlying member method code with example usage: obj.test_map_fields_as_expected(fields_to_map) without return types