mirror of
https://github.com/romanz/amodem.git
synced 2026-02-06 16:48:06 +08:00
PEP8 fixes
lambdas and coverage
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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()
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user