From c38208e10b4831c4ca371954bd5b7f656008f607 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Wed, 31 Dec 2014 10:58:54 +0200
Subject: [PATCH] calib: fix UT

---
 tests/test_calib.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/tests/test_calib.py b/tests/test_calib.py
index 0265087..0ae6539 100644
--- a/tests/test_calib.py
+++ b/tests/test_calib.py
@@ -37,16 +37,16 @@ def test_success():
 
 
 def test_errors():
-    p = ProcessMock()
-
-    def _write(data):
-        raise IOError()
-    p.write = _write
+    class WriteError(ProcessMock):
+        def write(self, data):
+            raise IOError()
+    p = WriteError()
     calib.send(config, p)
     assert p.buf.tell() == 0
 
-    def _read(data):
-        raise KeyboardInterrupt()
-    p.read = _read
+    class ReadError(ProcessMock):
+        def read(self, n):
+            raise KeyboardInterrupt()
+    p = ReadError()
     calib.recv(config, p, verbose=True)
     assert p.buf.tell() == 0