def test_make_dataset_from_variant_tensor_constructs_dataset(self):
with tf.Graph().as_default():
ds = graph_utils.make_dataset_from_variant_tensor(
tf.data.experimental.to_variant(tf.data.Dataset.range(5)), tf.int64)
self.assertIsInstance(ds, tf.compat.v2.data.Dataset)
result = ds.reduce(np.int64(0), lambda x, y: x + y)
with tf.Session() as sess:
self.assertEqual(sess.run(result), 10)
def test_make_dataset_from_variant_tensor_fails_with_bad_type(self):
with self.assertRaises(TypeError):
with tf.Graph().as_default():
graph_utils.make_dataset_from_variant_tensor(
tf.data.experimental.to_variant(tf.data.Dataset.range(5)),
'a')
def test_make_dataset_from_variant_tensor_fails_with_bad_tensor(self):
with self.assertRaises(TypeError):
with tf.Graph().as_default():
graph_utils.make_dataset_from_variant_tensor(
tf.constant(10), tf.int32)
