def test_compute(self):
var_pooling = pipeline.VarPool(3)
data_a = np.array([[1, 1], [2, 2], [3, 3], [4, 4], [5, 5]])
chunk_a = pipeline.Chunk(data_a, 0, False)
out_a = var_pooling.compute(chunk_a, 16000)
assert np.allclose(
out_a, np.array([[0.6666666666666666, 0.6666666666666666]]))
data_b = np.array([
[6, 1],
[7, 1],
[8, 1],
[9, 1],
])
chunk_b = pipeline.Chunk(data_b, 5, True)
out_b = var_pooling.compute(chunk_b, 16000)
assert np.allclose(
out_b,
np.array([[0.6666666666666666, 2.8888888888888893],
[0.6666666666666666, 0]]))
def test_compute_with_rest(self):
avg_pooling = pipeline.AvgPool(3)
data_a = np.array([[1, 1], [2, 2], [3, 3], [4, 4], [5, 5]])
chunk_a = pipeline.Chunk(data_a, 0, True)
out_a = avg_pooling.compute(chunk_a, 16000)
assert np.allclose(out_a, np.array([[2, 2], [4.5, 4.5]]))
def test_compute_with_left_frames(self):
data = np.array([[1, 2], [3, 4], [5, 6]])
chunk = pipeline.Chunk(data, offset=0, is_last=True)
step = pipeline.AddContext(left_frames=2, right_frames=0)
result = step.compute(chunk, 16000)
assert np.array_equal(
result,
np.array([[0, 0, 0, 0, 1, 2], [0, 0, 1, 2, 3, 4],
[1, 2, 3, 4, 5, 6]]))
def test_compute_with_rest(self):
var_pooling = pipeline.VarPool(3)
data_a = np.array([[1, 1], [2, 2], [3, 3], [4, 4], [5, 5]])
chunk_a = pipeline.Chunk(data_a, 0, True)
out_a = var_pooling.compute(chunk_a, 16000)
assert np.allclose(
out_a,
np.array([[0.6666666666666666, 0.6666666666666666], [0.25, 0.25]]))
def test_compute(self):
avg_pooling = pipeline.AvgPool(3)
data_a = np.array([[1, 1], [2, 2], [3, 3], [4, 4], [5, 5]])
chunk_a = pipeline.Chunk(data_a, 0, False)
out_a = avg_pooling.compute(chunk_a, 16000)
assert np.allclose(out_a, np.array([[2, 2]]))
data_b = np.array([
[6, 1],
[7, 1],
[8, 1],
[9, 1],
])
chunk_b = pipeline.Chunk(data_b, 5, True)
out_b = avg_pooling.compute(chunk_b, 16000)
assert np.allclose(out_b, np.array([[5, 10.0 / 3.0], [8, 1]]))
def test_compute_cleanup_after_one_utterance(self):
var_pooling = pipeline.VarPool(3)
# FIRST
data_a = np.array([[1, 1], [2, 2], [3, 3], [4, 4], [5, 5]])
chunk_a = pipeline.Chunk(data_a, 0, True)
out_a = var_pooling.compute(chunk_a, 16000)
assert np.allclose(
out_a,
np.array([[0.6666666666666666, 0.6666666666666666], [0.25, 0.25]]))
# SECOND
out_a = var_pooling.compute(chunk_a, 16000)
assert np.allclose(
out_a,
np.array([[0.6666666666666666, 0.6666666666666666], [0.25, 0.25]]))