def test_addition(): a = dsp_signal.Signal([1, 2, 3]) b = dsp_signal.Signal([4, 1, 2]) sum_signal = a + b assert (sum_signal.signal[0, 0] == 5) assert (sum_signal.signal[0, 1] == 3) assert (sum_signal.signal[0, 2] == 5)
def test_mult(): a = dsp_signal.Signal([1, 2, 3]) b = dsp_signal.Signal([4, 1, 2]) mult_signal = a * b assert (mult_signal.signal[0, 0] == 4) assert (mult_signal.signal[0, 1] == 2) assert (mult_signal.signal[0, 2] == 6)
def test_concatenate(): a = dsp_signal.Signal([1, 2, 3]) a = a.concatenate(a) assert (a.signal[0, 0] == 1) assert (a.signal[0, 1] == 2) assert (a.signal[0, 2] == 3) assert (a.signal[0, 3] == 1) assert (a.signal[0, 4] == 2) assert (a.signal[0, 5] == 3)
def test_shift(): a = dsp_signal.Signal([1, 2, 3]) a = a.shift(1) assert (a.signal[0, 0] == 0) assert (a.signal[0, 1] == 1) assert (a.signal[0, 2] == 2) a = a.shift(-2) assert (a.signal[0, 0] == 2) assert (a.signal[0, 1] == 0) assert (a.signal[0, 2] == 0)
def test_clip(): a = dsp_signal.Signal([1, 2, 3]) a = a.clip(1, 2) assert (a.signal[0, 0] == 2)
def test_scalar_mult(): a = dsp_signal.Signal([1, 2, 3]) * 2 assert (a.signal[0, 0] == 2) assert (a.signal[0, 1] == 4) assert (a.signal[0, 2] == 6)
def test_scalar_addition(): a = dsp_signal.Signal([1, 2, 3]) + 6 assert (a.signal[0, 0] == 7) assert (a.signal[0, 1] == 8) assert (a.signal[0, 2] == 9)