From 34a892e72cc6ad7b96694cc62610814bda58ac42 Mon Sep 17 00:00:00 2001 From: Roman Zeyde Date: Thu, 19 Feb 2015 15:20:42 +0200 Subject: [PATCH] equalizer: remove unused dependencies --- tests/test_equalizer.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_equalizer.py b/tests/test_equalizer.py index 3811887..b95c6f9 100644 --- a/tests/test_equalizer.py +++ b/tests/test_equalizer.py @@ -1,16 +1,17 @@ -from numpy.linalg import norm from numpy.random import RandomState import numpy as np import utils from amodem import equalizer -from amodem import levinson +from amodem import dsp from amodem import config config = config.fastest() def assert_approx(x, y, e=1e-12): - assert norm(x - y) < e * norm(x) + x = x.flatten() + y = y.flatten() + assert dsp.norm(x - y) < e * dsp.norm(x) def test_training(): @@ -57,7 +58,7 @@ def test_signal(): lookahead = 2 h = equalizer.train( signal=y, expected=x, order=len(den), lookahead=lookahead) - assert norm(h[:lookahead]) < 1e-12 + assert dsp.norm(h[:lookahead]) < 1e-12 h = h[lookahead:] assert_approx(h, den / num)