def testTextSummary(self):
with self.cached_session():
with self.assertRaises(ValueError):
num = array_ops.constant(1)
summary_lib.text('foo', num)
# The API accepts vectors.
arr = array_ops.constant(['one', 'two', 'three'])
summ = summary_lib.text('foo', arr)
self.assertEqual(summ.op.type, 'TensorSummaryV2')
# the API accepts scalars
summ = summary_lib.text('foo', array_ops.constant('one'))
self.assertEqual(summ.op.type, 'TensorSummaryV2')
def testTextSummary(self):
with self.cached_session():
with self.assertRaises(ValueError):
num = array_ops.constant(1)
summary_lib.text('foo', num)
# The API accepts vectors.
arr = array_ops.constant(['one', 'two', 'three'])
summ = summary_lib.text('foo', arr)
self.assertEqual(summ.op.type, 'TensorSummaryV2')
# the API accepts scalars
summ = summary_lib.text('foo', array_ops.constant('one'))
self.assertEqual(summ.op.type, 'TensorSummaryV2')
def test_text_summary_v2(self):
"""Tests text v2 invocation."""
with test.mock.patch.object(summary_v2, 'text',
autospec=True) as mock_text_v2:
with summary_ops_v2.create_summary_file_writer(
self.get_temp_dir()).as_default(step=22):
i = constant_op.constant('lorem ipsum', dtype=dtypes.string)
tensor = summary_lib.text('text', i)
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_text_v2.assert_called_once_with('text', data=i, step=22)
def test_text(self):
op = summary_lib.text('lorem_ipsum', constant_op.constant('dolor sit amet'))
value = self._value_from_op(op)
assert value.HasField('tensor'), value
self._assert_noop(value)