From 0d52546d815a0709b6808dad3b000fdd61c19c0e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A1s=20Veres-Szentkir=C3=A1lyi?= Date: Thu, 12 Jan 2017 12:26:01 +0100 Subject: [PATCH] test_sstv: select open namespace depending on Python major version --- pysstv/tests/test_sstv.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pysstv/tests/test_sstv.py b/pysstv/tests/test_sstv.py index 10162c5..337a049 100644 --- a/pysstv/tests/test_sstv.py +++ b/pysstv/tests/test_sstv.py @@ -5,6 +5,7 @@ import pickle import mock from mock import MagicMock from six import StringIO +from six import PY2 import hashlib from pysstv import sstv @@ -70,7 +71,8 @@ class TestSSTV(unittest.TestCase): sio = StringIO() sio.close = MagicMock() # ignore close() so we can .getvalue() mock_open = MagicMock(return_value=sio) - with mock.patch('__builtin__.open', mock_open): + ns = '__builtin__' if PY2 else 'io' + with mock.patch('{0}.open'.format(ns), mock_open): self.s.write_wav('unittest.wav') expected = 'dd7eed880ab3360fb79ce09c469deee2' data = sio.getvalue()