code: def test_non_nested(self):
      d = {"a": 1, "b": 2}
      iterator = tff.utils.iterate_nested(d)
      self._assert_keys_values(
          iterator, expected_keys=["a", "b"], expected_values=[1, 2]
      )
  
file_path: tf-quant-finance/tf_quant_finance/utils/tf_functions_test.py
imports: from typing import Dict
  import dataclasses
  import tensorflow.compat.v2
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 2 for calling the TfFunctionsTest obj's underlying member method code with example usage: obj.test_non_nested() without return types