code: def test_multiplex_float(self):
      a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0])
      b = tf.constant([10.0, 20.0, 30.0, 40.0, 50.0])
      cond = tf.constant([True, False, True, False, True], dtype=bool)
      # expected result is [1.0, 20.0, 3.0, 40.0, 5.0]
      expect = np.where(
          self.evaluate(cond), self.evaluate(a), self.evaluate(b)
      )
      result = multiplex_2_op.multiplex(cond, a, b)
      self.assertAllEqual(result, expect)
  
file_path: tensorflow/tensorflow/examples/custom_ops_doc/multiplex_2/multiplex_2_test.py
imports: import numpy
  import tensorflow
  from tensorflow.examples.custom_ops_doc.multiplex_2 import multiplex_2_op
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 2 for calling the MultiplexOpRank1Test obj's underlying member method code with example usage: obj.test_multiplex_float() without return types