mirror of
https://github.com/dnet/pySSTV.git
synced 2026-01-06 08:40:29 +01:00
test_sstv: select open namespace depending on Python major version
This commit is contained in:
parent
ac0f89bfdf
commit
0d52546d81
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Reference in a new issue