def test_get():
signal = Signal(["test", "test2"], [[1, 2], [3, 4]])
audio = signal.get("test")
assert audio == [1, 2]
with pytest.raises(ValueError) as pe:
audio = signal.get("test3")
def test_overlay():
dummy_signal = Signal(["test"], [np.array(range(100), dtype=np.float32)])
augment = Augment()
augment.overlay((5, 10), 17, 1)
augment.augment(dummy_signal, "test")
audio = dummy_signal.get("test")
actual = np.array([11, 12, 13, 14, 15], dtype=np.float32)
assert np.allclose(audio[17: 22], actual)