| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879 |
- # Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
- # For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
- # Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt
- """Hooks for nose library."""
- import re
- import textwrap
- import astroid.builder
- from astroid.brain.helpers import register_module_extender
- from astroid.exceptions import InferenceError
- from astroid.manager import AstroidManager
- _BUILDER = astroid.builder.AstroidBuilder(AstroidManager())
- CAPITALS = re.compile("([A-Z])")
- def _pep8(name, caps=CAPITALS):
- 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 (InferenceError, StopIteration):
- 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
- register_module_extender(
- AstroidManager(), "nose.tools.trivial", _nose_tools_trivial_transform
- )
- AstroidManager().register_transform(
- astroid.Module, _nose_tools_transform, lambda n: n.name == "nose.tools"
- )
|
