def test_summarywriter_clipped_audio(self): log_dir = tempfile.mkdtemp() summary_writer = SummaryWriter(log_dir=log_dir) expected_audio_samples = onp.random.uniform(low=-2., high=2., size=(2, 48000, 2)) summary_writer.audio(tag='audio_test', audiodata=expected_audio_samples, step=1, max_outputs=1) summary_value = self.parse_and_return_summary_value(path=log_dir) self.assertEqual(summary_value.tag, 'audio_test') # Assert one audio files is parsed. self.assertLen(summary_value.tensor.string_val, 1) # actual_audio is clipped. actual_audio = tf.audio.decode_wav( summary_value.tensor.string_val[0]).audio self.assertFalse( onp.allclose(expected_audio_samples[0], actual_audio, atol=1e-04)) clipped_audio = onp.clip(onp.array(expected_audio_samples[0]), -1, 1) self.assertTrue(onp.allclose(clipped_audio, actual_audio, atol=1e-04))
def test_summarywriter_audio(self): log_dir = tempfile.mkdtemp() summary_writer = SummaryWriter(log_dir=log_dir) expected_audio_samples = onp.random.uniform(low=-1., high=1., size=(2, 48000, 2)) summary_writer.audio(tag='audio_test', audiodata=expected_audio_samples, step=1) summary_value = self.parse_and_return_summary_value(path=log_dir) self.assertEqual(summary_value.tag, 'audio_test') # Assert two audio files are parsed. self.assertLen(summary_value.tensor.string_val, 2) # Assert values. actual_audio_1 = tf.audio.decode_wav( summary_value.tensor.string_val[0]).audio self.assertTrue( onp.allclose(expected_audio_samples[0], actual_audio_1, atol=1e-04)) actual_audio_2 = tf.audio.decode_wav( summary_value.tensor.string_val[1]).audio self.assertTrue( onp.allclose(expected_audio_samples[1], actual_audio_2, atol=1e-04))
def test_summarywriter_audio_sampled_output(self): log_dir = tempfile.mkdtemp() summary_writer = SummaryWriter(log_dir=log_dir) expected_audio_samples = onp.random.uniform( low=-1., high=1., size=(2, 48000, 2)) summary_writer.audio( tag='audio_test', audiodata=expected_audio_samples, step=1, max_outputs=1) summary_value = self.parse_and_return_summary_value(path=log_dir) self.assertEqual(summary_value.tag, 'audio_test') # Assert only the first audio clip is available. self.assertLen(summary_value.tensor.string_val, 2) # Assert values. actual_audio = tf.audio.decode_wav(summary_value.tensor.string_val[0]).audio self.assertTrue(onp.allclose( expected_audio_samples[0], actual_audio, atol=1e-04))