# -*- coding: utf-8 -*-
try:
import unittest2 as unittest
except ImportError:
import unittest
try:
from unittest import mock
except ImportError:
import mock
from bpython.translations import init
from bpython._py3compat import py3
from six.moves import builtins
class FixLanguageTestCase(unittest.TestCase):
@classmethod
def setUpClass(cls):
init(languages=['en'])
class MagicIterMock(mock.MagicMock):
if py3:
__next__ = mock.Mock(return_value=None)
else:
next = mock.Mock(return_value=None)
def builtin_target(obj):
"""Returns mock target string of a builtin"""
return '%s.%s' % (builtins.__name__, obj.__name__)