class_docstr: Tests for `split_dict_py_tf` and `merge_dicts` methods.
class_docstr_tok: ["Tests","for","`","split_dict_py_tf","`","and","`","merge_dicts","`","methods","."]
docstr: Tests that `split_dict_py_tf` works with flat dictionary.
docstr_tok: ["Tests","that","`","split_dict_py_tf","`","works","with","flat","dictionary","."]
code: def test_split_dict_py_tf_basic(self):
      """Tests that `split_dict_py_tf` works with flat dictionary."""
      const = tf.constant(2.0)
      test_dict = {"py": 1.0, "tf": const}
      expected_d_py = {"py": 1.0}
      expected_d_tf = {"tf": const}
      d_py, d_tf = py_utils.split_dict_py_tf(test_dict)
      self.assertDictEqual(expected_d_py, d_py)
      self.assertDictEqual(expected_d_tf, d_tf)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.utils import py_utils
label_desc: Use image node_id 2 for calling the SplitMergeDictTest obj's underlying member method code with example usage: obj.test_split_dict_py_tf_basic() without return types