Skip to content

Commit a34d741

Browse files
Fix builtin module reference for mock in Python 3
1 parent af85ee2 commit a34d741

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

test/util/test_backend.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,12 @@
44
import mock
55
from autosklearn.util.backend import Backend
66

7+
from sys import version_info
8+
if version_info.major == 2:
9+
import __builtin__ as builtins
10+
else:
11+
import builtins
12+
713

814
class BackendModelsTest(unittest.TestCase):
915

@@ -18,7 +24,7 @@ def setUp(self):
1824
self.backend.get_model_dir = lambda: self.model_directory
1925

2026
@mock.patch('six.moves.cPickle.load')
21-
@mock.patch('__builtin__.open')
27+
@mock.patch.object(builtins, 'open')
2228
def test_loads_model_by_seed_and_id(self, openMock, pickleLoadMock):
2329
seed = 13
2430
idx = 17
@@ -29,7 +35,7 @@ def test_loads_model_by_seed_and_id(self, openMock, pickleLoadMock):
2935
self.assertEqual(expected_model, actual_model)
3036

3137
@mock.patch('six.moves.cPickle.load')
32-
@mock.patch('__builtin__.open')
38+
@mock.patch.object(builtins, 'open')
3339
def test_loads_models_by_identifiers(self, openMock, pickleLoadMock):
3440
seed = 13
3541
idx = 17

0 commit comments

Comments
 (0)