bpython/bpython/test/__init__.py at master · jsjohnst/bpython · GitHub

35 lines (24 loc) · 688 Bytes

# -*- 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__)

Expand file tree