code: def test_where_fallback(self):
      self.assertAllEqual(
          [1.0, 0.0],
          ps.smart_where(
              tf.constant([True, False]),
              lambda: tf.ones([]),
              lambda: tf.zeros([]),
          ),
      )
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 5 for calling the SmartWhereTest obj's underlying member method code with example usage: obj.test_where_fallback() without return types