From 97e992ea567a8a5394e17a9a3e9eb9bb3c28b7f6 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Wed, 18 Feb 2015 18:14:41 +0200
Subject: [PATCH] audio: return self from Interface.load()

---
 amodem/audio.py     | 1 +
 tests/test_audio.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/amodem/audio.py b/amodem/audio.py
index 62a4069..eafc728 100644
--- a/amodem/audio.py
+++ b/amodem/audio.py
@@ -17,6 +17,7 @@ class Interface(object):
         assert self._error_string(0) == b'Success'
         version = self.call('GetVersionText', restype=ctypes.c_char_p)
         log.info('%s loaded', version)
+        return self
 
     def _error_string(self, code):
         return self.call('GetErrorText', code, restype=ctypes.c_char_p)
diff --git a/tests/test_audio.py b/tests/test_audio.py
index f4d1620..cad185d 100644
--- a/tests/test_audio.py
+++ b/tests/test_audio.py
@@ -15,7 +15,7 @@ def test():
         lib.Pa_OpenStream.return_value = 0
         cdll.return_value = lib
         interface = audio.Interface(config=config.fastest(), debug=True)
-        interface.load(name='portaudio')
+        assert interface.load(name='portaudio') is interface
         with interface:
             s = interface.player()
             assert s.params.device == 1