Gecko [ lorencats.com ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
brain_argparse.py	1 KB	-rw-r--r--
brain_attrs.py	1.86 KB	-rw-r--r--
brain_builtin_inference.py	26.28 KB	-rw-r--r--
brain_collections.py	2.76 KB	-rw-r--r--
brain_curses.py	3.23 KB	-rw-r--r--
brain_dateutil.py	795 B	-rw-r--r--
brain_fstrings.py	2.07 KB	-rw-r--r--
brain_functools.py	6.16 KB	-rw-r--r--
brain_gi.py	6.65 KB	-rw-r--r--
brain_hashlib.py	2.11 KB	-rw-r--r--
brain_io.py	1.44 KB	-rw-r--r--
brain_mechanize.py	889 B	-rw-r--r--
brain_multiprocessing.py	3.04 KB	-rw-r--r--
brain_namedtuple_enum.py	13.7 KB	-rw-r--r--
brain_nose.py	2.16 KB	-rw-r--r--
brain_numpy.py	14.44 KB	-rw-r--r--
brain_pkg_resources.py	2.21 KB	-rw-r--r--
brain_pytest.py	2.33 KB	-rw-r--r--
brain_qt.py	2.38 KB	-rw-r--r--
brain_random.py	2.38 KB	-rw-r--r--
brain_re.py	1.1 KB	-rw-r--r--
brain_six.py	6 KB	-rw-r--r--
brain_ssl.py	3.47 KB	-rw-r--r--
brain_subprocess.py	3.55 KB	-rw-r--r--
brain_threading.py	767 B	-rw-r--r--
brain_typing.py	2.66 KB	-rw-r--r--
brain_uuid.py	564 B	-rw-r--r--

Code Editor : brain_nose.py

# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
# For details: https://github.com/PyCQA/astroid/blob/master/COPYING.LESSER

"""Hooks for nose library."""

import re
import textwrap

import astroid
import astroid.builder

_BUILDER = astroid.builder.AstroidBuilder(astroid.MANAGER)

def _pep8(name, caps=re.compile("([A-Z])")):
    return caps.sub(lambda m: "_" + m.groups()[0].lower(), name)

def _nose_tools_functions():
    """Get an iterator of names and bound methods."""
    module = _BUILDER.string_build(
        textwrap.dedent(
            """
    import unittest

class Test(unittest.TestCase):
        pass
    a = Test()
    """
        )
    )
    try:
        case = next(module["a"].infer())
    except astroid.InferenceError:
        return
    for method in case.methods():
        if method.name.startswith("assert") and "_" not in method.name:
            pep8_name = _pep8(method.name)
            yield pep8_name, astroid.BoundMethod(method, case)
        if method.name == "assertEqual":
            # nose also exports assert_equals.
            yield "assert_equals", astroid.BoundMethod(method, case)

def _nose_tools_transform(node):
    for method_name, method in _nose_tools_functions():
        node.locals[method_name] = [method]

def _nose_tools_trivial_transform():
    """Custom transform for the nose.tools module."""
    stub = _BUILDER.string_build("""__all__ = []""")
    all_entries = ["ok_", "eq_"]

for pep8_name, method in _nose_tools_functions():
        all_entries.append(pep8_name)
        stub[pep8_name] = method

# Update the __all__ variable, since nose.tools
    # does this manually with .append.
    all_assign = stub["__all__"].parent
    all_object = astroid.List(all_entries)
    all_object.parent = all_assign
    all_assign.value = all_object
    return stub

astroid.register_module_extender(
    astroid.MANAGER, "nose.tools.trivial", _nose_tools_trivial_transform
)
astroid.MANAGER.register_transform(
    astroid.Module, _nose_tools_transform, lambda n: n.name == "nose.tools"
)

${this.title}

Code Editor : brain_nose.py