code: def test_np_map_fn():
      # Scalar to scalar.
      add_ten = lambda i: i + 10
      np.testing.assert_array_equal(
          np_utils.np_map_fn(add_ten, np.array([0, 1, 2])),
          np.array([10, 11, 12]),
      )
  
      # Dicts to scalar.
      operation_on_dict = (
          lambda d: d["a"] + d["b"] if d["a"] % 2 == 0 else 0
      )
      result = np_utils.np_map_fn(
          operation_on_dict,
          {
              "a": np.array([0, 1, 2]),
              "b": np.array([5, 6, 7]),
          },
      )
      expected = np.array([5, 0, 9])
      np.testing.assert_array_equal(result, expected)
  
      # Dicts to dicts.
      operation_on_dict = lambda _: {"c": 1, "d": 2}
      result = np_utils.np_map_fn(
          operation_on_dict,
          {
              "a": np.array([0, 1, 2]),
              "b": np.array([5, 6, 7]),
          },
      )
      expected = {
          "c": np.array([1, 1, 1]),
          "d": np.array([2, 2, 2]),
      }
      assert len(result) == 2
      np.testing.assert_array_equal(result["c"], expected["c"])
      np.testing.assert_array_equal(result["d"], expected["d"])
  
label_desc: Use image node_id 2 for calling a global function with example usage: test_np_map_fn() without return types