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))
