experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Tests for TF based value range picking

Browse Source
This commit is contained in:
Tilman Kamp 2020-06-16 13:32:32 +02:00
parent 5b6de213d8
commit 2d5dcc359a
1 changed files with 42 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
tests/test_value_range.py
View File

@ -1,6 +1,8 @@
import unittest
from deepspeech_training.util.helpers import ValueRange, get_value_range, pick_value_from_range
import numpy as np
import tensorflow as tf
from deepspeech_training.util.helpers import ValueRange, get_value_range, pick_value_from_range, tf_pick_value_from_range
class TestValueRange(unittest.TestCase):
@ -56,12 +58,23 @@ class TestValueRange(unittest.TestCase):
class TestPickValueFromFixedRange(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestPickValueFromFixedRange, self).__init__(*args, **kwargs)
self.session = tf.Session()
self.clock_ph = tf.placeholder(dtype=tf.float64, name='clock')
def _ending_tester(self, value_range, clock, expected):
is_int = isinstance(value_range.start, int)
result = pick_value_from_range(value_range, clock)
self.assertEqual(result, expected)
self.assertTrue(isinstance(result, int if is_int else float))
with tf.Session() as session:
tf_pick = tf_pick_value_from_range(value_range, clock=self.clock_ph)
def run_pick(_, c):
return session.run(tf_pick, feed_dict={self.clock_ph: c})
is_int = isinstance(value_range.start, int)
for pick, int_type, float_type in [(pick_value_from_range, int, float), (run_pick, np.int32, np.float32)]:
result = pick(value_range, clock)
self.assertEqual(result, expected)
self.assertTrue(isinstance(result, int_type if is_int else float_type))
def test_int_0(self):
self._ending_tester(ValueRange(1, 3, 0), 0.0, 1)
@ -83,31 +96,43 @@ class TestPickValueFromFixedRange(unittest.TestCase):
class TestPickValueFromRandomizedRange(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestPickValueFromRandomizedRange, self).__init__(*args, **kwargs)
self.session = tf.Session()
self.clock_ph = tf.placeholder(dtype=tf.float64, name='clock')
def _ending_tester(self, value_range, clock, expected_min, expected_max):
is_int = isinstance(value_range.start, int)
results = list(map(lambda x: pick_value_from_range(value_range, clock), range(100)))
self.assertGreater(len(set(results)), 80)
self.assertTrue(all(map(lambda x: expected_min <= x <= expected_max, results)))
self.assertTrue(all(map(lambda x: isinstance(x, int if is_int else float), results)))
def _ending_tester(self, value_range, clock_min, clock_max, expected_min, expected_max):
with self.session as session:
tf_pick = tf_pick_value_from_range(value_range, clock=self.clock_ph)
def run_pick(_, c):
return session.run(tf_pick, feed_dict={self.clock_ph: c})
is_int = isinstance(value_range.start, int)
clock_range = np.arange(clock_min, clock_max, (clock_max - clock_min) / 100.0)
for pick, int_type, float_type in [(pick_value_from_range, int, float), (run_pick, np.int32, np.float32)]:
results = [pick(value_range, c) for c in clock_range]
self.assertGreater(len(set(results)), 80)
self.assertTrue(all(map(lambda x: expected_min <= x <= expected_max, results)))
self.assertTrue(all(map(lambda x: isinstance(x, int_type if is_int else float_type), results)))
def test_int_0(self):
self._ending_tester(ValueRange(10000, 30000, 10000), 0.0, 0, 20000)
self._ending_tester(ValueRange(10000, 30000, 10000), 0.0, 0.1, 0, 22000)
def test_int_half(self):
self._ending_tester(ValueRange(10000, 30000, 10000), 0.5, 10000, 30000)
self._ending_tester(ValueRange(10000, 30000, 10000), 0.4, 0.6, 8000, 32000)
def test_int_1(self):
self._ending_tester(ValueRange(10000, 30000, 10000), 1.0, 20000, 40000)
self._ending_tester(ValueRange(10000, 30000, 10000), 0.8, 1.0, 16000, 40000)
def test_float_0(self):
self._ending_tester(ValueRange(10000.0, 30000.0, 10000.0), 0.0, 0.0, 20000.0)
self._ending_tester(ValueRange(10000.0, 30000.0, 10000.0), 0.0, 0.1, 0.0, 22000.0)
def test_float_half(self):
self._ending_tester(ValueRange(10000.0, 30000.0, 10000.0), 0.5, 10000.0, 30000.0)
self._ending_tester(ValueRange(10000.0, 30000.0, 10000.0), 0.4, 0.6, 8000.0, 32000.0)
def test_float_1(self):
self._ending_tester(ValueRange(10000.0, 30000.0, 10000.0), 1.0, 20000.0, 40000.0)
self._ending_tester(ValueRange(10000.0, 30000.0, 10000.0), 0.8, 1.0, 16000.0, 40000.0)
if __name__ == '__main__':