PEP8 fixes

lambdas and coverage
This commit is contained in:
Roman Zeyde
2015-02-07 08:52:27 +02:00
parent 5b94d7fd49
commit a2b220c8e4
4 changed files with 18 additions and 4 deletions

View File

@@ -72,6 +72,10 @@ def norm(x):
return np.sqrt(np.dot(x.conj(), x).real)
def rms(x):
return np.mean(np.abs(x) ** 2, axis=0) ** 0.5
def coherence(x, omega):
n = len(x)
Hc = exp_iwt(-omega, n) / np.sqrt(0.5*n)

View File

@@ -79,10 +79,9 @@ class Receiver(object):
error_rate = errors.sum() / errors.size
errors = np.array(symbols - train_symbols)
rms = lambda x: (np.mean(np.abs(x) ** 2, axis=0) ** 0.5)
noise_rms = rms(errors)
signal_rms = rms(train_symbols)
noise_rms = dsp.rms(errors)
signal_rms = dsp.rms(train_symbols)
SNRs = 20.0 * np.log10(signal_rms / noise_rms)
self.plt.figure()

View File

@@ -1,4 +1,5 @@
from amodem import common
from amodem import config
import numpy as np
@@ -59,3 +60,11 @@ def test_holder():
assert a.x == d['x']
assert a.y == d['y']
assert repr(a) == 'AttributeHolder(x=1, y=2.3)'
def test_configs():
default = config.Configuration()
fastest = config.fastest()
slowest = config.slowest()
assert slowest.modem_bps <= default.modem_bps
assert fastest.modem_bps >= default.modem_bps

View File

@@ -48,7 +48,9 @@ def test_qam():
decoded = list(q.decode(S))
assert decoded == bits
noise = lambda A: A*(r.uniform(-1, 1) + 1j*r.uniform(-1, 1))
def noise(A):
return A*(r.uniform(-1, 1) + 1j*r.uniform(-1, 1))
noised_symbols = [(s + noise(1e-3)) for s in S]
decoded = list(q.decode(noised_symbols))
assert decoded == bits