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)