mirror of
https://github.com/romanz/amodem.git
synced 2026-04-20 13:16:42 +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)
|
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):
|
def coherence(x, omega):
|
||||||
n = len(x)
|
n = len(x)
|
||||||
Hc = exp_iwt(-omega, n) / np.sqrt(0.5*n)
|
Hc = exp_iwt(-omega, n) / np.sqrt(0.5*n)
|
||||||
|
|||||||
@@ -79,10 +79,9 @@ class Receiver(object):
|
|||||||
error_rate = errors.sum() / errors.size
|
error_rate = errors.sum() / errors.size
|
||||||
|
|
||||||
errors = np.array(symbols - train_symbols)
|
errors = np.array(symbols - train_symbols)
|
||||||
rms = lambda x: (np.mean(np.abs(x) ** 2, axis=0) ** 0.5)
|
|
||||||
|
|
||||||
noise_rms = rms(errors)
|
noise_rms = dsp.rms(errors)
|
||||||
signal_rms = rms(train_symbols)
|
signal_rms = dsp.rms(train_symbols)
|
||||||
SNRs = 20.0 * np.log10(signal_rms / noise_rms)
|
SNRs = 20.0 * np.log10(signal_rms / noise_rms)
|
||||||
|
|
||||||
self.plt.figure()
|
self.plt.figure()
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
from amodem import common
|
from amodem import common
|
||||||
|
from amodem import config
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
|
|
||||||
@@ -59,3 +60,11 @@ def test_holder():
|
|||||||
assert a.x == d['x']
|
assert a.x == d['x']
|
||||||
assert a.y == d['y']
|
assert a.y == d['y']
|
||||||
assert repr(a) == 'AttributeHolder(x=1, y=2.3)'
|
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))
|
decoded = list(q.decode(S))
|
||||||
assert decoded == bits
|
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]
|
noised_symbols = [(s + noise(1e-3)) for s in S]
|
||||||
decoded = list(q.decode(noised_symbols))
|
decoded = list(q.decode(noised_symbols))
|
||||||
assert decoded == bits
|
assert decoded == bits
|
||||||
|
|||||||
Reference in New Issue
Block a user