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

39 lines (26 loc) · 769 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

import os

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

TEST_CONFIG = os.path.join(os.path.dirname(__file__), "test.config")

Expand file tree