def test_scalar_summary_v2__global_step_not_set(self):
"""Tests scalar v2 invocation when global step is not set."""
with self.assertWarnsRegex(UserWarning, 'global step not set'):
with test.mock.patch.object(
summary_v2, 'scalar', autospec=True) as mock_scalar_v2:
with summary_ops_v2.create_summary_file_writer(
'/tmp/test').as_default():
summary_lib.scalar('float', constant_op.constant(2.5))
mock_scalar_v2.assert_not_called()
def test_scalar_summary_v2__w_writer(self):
"""Tests scalar v2 invocation with a v2 writer."""
with test.mock.patch.object(summary_v2, 'scalar') as mock_scalar_v2:
with summary_ops_v2.create_summary_file_writer('/tmp/test').as_default(
step=1):
i = constant_op.constant(2.5)
tensor = summary_lib.scalar('float', i)
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_scalar_v2.assert_called_once_with('float', data=i)
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_scalar_summary_v2__family(self):
"""Tests `family` arg handling when scalar v2 is invoked."""
with test.mock.patch.object(
summary_v2, 'scalar', autospec=True) as mock_scalar_v2:
with summary_ops_v2.create_summary_file_writer('/tmp/test').as_default(
step=1):
tensor = summary_lib.scalar(
'float', constant_op.constant(2.5), family='otter')
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_scalar_v2.assert_called_once_with(
'otter/otter/float', data=constant_op.constant(2.5), step=1)
def test_scalar_summary_v2__v1_set_step(self):
"""Tests scalar v2 invocation when v1 step is set."""
global_step = training_util.create_global_step()
global_step.assign(1024)
with test.mock.patch.object(
summary_v2, 'scalar', autospec=True) as mock_scalar_v2:
with summary_ops_v2.create_summary_file_writer('/tmp/test').as_default():
i = constant_op.constant(2.5)
tensor = summary_lib.scalar('float', i)
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_scalar_v2.assert_called_once_with('float', data=i, step=1024)
def test_histogram_summary_v2(self):
"""Tests histogram v2 invocation."""
with test.mock.patch.object(
summary_v2, 'histogram', autospec=True) as mock_histogram_v2:
with summary_ops_v2.create_summary_file_writer(
self.get_temp_dir()).as_default(step=3):
i = array_ops.ones((1024,))
tensor = summary_lib.histogram('histogram', i, family='family')
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_histogram_v2.assert_called_once_with(
'family/family/histogram', data=i, step=3)
def test_scalar_summary_v2__family_w_outer_scope(self):
"""Tests `family` arg handling when there is an outer scope."""
with test.mock.patch.object(
summary_v2, 'scalar', autospec=True) as mock_scalar_v2:
with summary_ops_v2.create_summary_file_writer('/tmp/test').as_default(
step=1):
with ops.name_scope_v2('sea'):
tensor = summary_lib.scalar(
'float', constant_op.constant(3.5), family='crabnet')
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_scalar_v2.assert_called_once_with(
'crabnet/sea/crabnet/float', data=constant_op.constant(3.5), step=1)
def test_image_summary_v2(self):
"""Tests image v2 invocation."""
with test.mock.patch.object(
summary_v2, 'image', autospec=True) as mock_image_v2:
with summary_ops_v2.create_summary_file_writer('/tmp/test').as_default(
step=2):
i = array_ops.ones((5, 4, 4, 3))
with ops.name_scope_v2('outer'):
tensor = summary_lib.image('image', i, max_outputs=3, family='family')
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_image_v2.assert_called_once_with(
'family/outer/family/image', data=i, step=2, max_outputs=3)
def test_audio_summary_v2(self):
"""Tests audio v2 invocation."""
with test.mock.patch.object(
summary_v2, 'audio', autospec=True) as mock_audio_v2:
with summary_ops_v2.create_summary_file_writer(
self.get_temp_dir()).as_default(step=10):
i = array_ops.ones((5, 3, 4))
with ops.name_scope_v2('dolphin'):
tensor = summary_lib.audio('wave', i, 0.2, max_outputs=3)
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_audio_v2.assert_called_once_with(
'dolphin/wave', data=i, sample_rate=0.2, step=10, max_outputs=3)
def test_audio_summary_v2__2d_tensor(self):
"""Tests audio v2 invocation with 2-D tensor input."""
with test.mock.patch.object(
summary_v2, 'audio', autospec=True) as mock_audio_v2:
with summary_ops_v2.create_summary_file_writer(
self.get_temp_dir()).as_default(step=11):
input_2d = array_ops.ones((5, 3))
tensor = summary_lib.audio('wave', input_2d, 0.2, max_outputs=3)
# Returns empty string.
self.assertEqual(tensor.numpy(), b'')
self.assertEqual(tensor.dtype, dtypes.string)
mock_audio_v2.assert_called_once_with(
'wave', data=test.mock.ANY, sample_rate=0.2, step=11, max_outputs=3)
input_3d = array_ops.ones((5, 3, 1)) # 3-D input tensor
self.assertAllEqual(mock_audio_v2.call_args[1]['data'], input_3d)
