test calib.py

amodem/calib.py

@@ -32,11 +32,7 @@ def recv(wave_record=wave.record, reporter=sys.stdout.write):
             data = p.stdout.read(len(sig_dump))
             if len(data) < len(sig_dump):
                 return
-            try:
-                x = common.loads(data)
-            except common.SaturationError as e:
-                print('saturation: {}'.format(e))
-                continue
+            x = common.loads(data)
             x = x - np.mean(x)
 
             normalization_factor = np.sqrt(0.5 * len(x)) * dsp.norm(x)

tests/test_calib.py

@@ -1,6 +1,7 @@
-from amodem import calib
+from amodem import calib, config, common
 
 from io import BytesIO
+import numpy as np
 
 
 class ProcessMock(object):
@@ -33,3 +34,19 @@ def test():
     calib.send(p)
     p.buf.seek(0)
     calib.recv(p, reporter=verify)
+
+
+def test_errors():
+    p = ProcessMock()
+
+    def _write(data):
+        raise IOError()
+    p.write = _write
+    calib.send(p)
+    assert p.buf.tell() == 0
+
+    def _read(data):
+        raise KeyboardInterrupt()
+    p.read = _read
+    calib.recv(p)
+    assert p.buf.tell() == 0