123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293 |
- # Copyright (c) 2014 Google, Inc.
- # Copyright (c) 2014-2016 Claudiu Popa <pcmanticore@gmail.com>
-
- # 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
-
- import inspect
- import os
- import unittest
-
- from six.moves import builtins
-
- from astroid.builder import AstroidBuilder, extract_node
- from astroid.raw_building import (
- attach_dummy_node, build_module,
- build_class, build_function, build_from_import
- )
- from astroid import test_utils
- from astroid import nodes
- from astroid.bases import BUILTINS
-
-
- class RawBuildingTC(unittest.TestCase):
-
- def test_attach_dummy_node(self):
- node = build_module('MyModule')
- attach_dummy_node(node, 'DummyNode')
- self.assertEqual(1, len(list(node.get_children())))
-
- def test_build_module(self):
- node = build_module('MyModule')
- self.assertEqual(node.name, 'MyModule')
- self.assertEqual(node.pure_python, False)
- self.assertEqual(node.package, False)
- self.assertEqual(node.parent, None)
-
- def test_build_class(self):
- node = build_class('MyClass')
- self.assertEqual(node.name, 'MyClass')
- self.assertEqual(node.doc, None)
-
- def test_build_function(self):
- node = build_function('MyFunction')
- self.assertEqual(node.name, 'MyFunction')
- self.assertEqual(node.doc, None)
-
- def test_build_function_args(self):
- args = ['myArgs1', 'myArgs2']
- # pylint: disable=no-member; not aware of postinit
- node = build_function('MyFunction', args)
- self.assertEqual('myArgs1', node.args.args[0].name)
- self.assertEqual('myArgs2', node.args.args[1].name)
- self.assertEqual(2, len(node.args.args))
-
- def test_build_function_defaults(self):
- # pylint: disable=no-member; not aware of postinit
- defaults = ['defaults1', 'defaults2']
- node = build_function('MyFunction', None, defaults)
- self.assertEqual(2, len(node.args.defaults))
-
- def test_build_from_import(self):
- names = ['exceptions, inference, inspector']
- node = build_from_import('astroid', names)
- self.assertEqual(len(names), len(node.names))
-
- @test_utils.require_version(minver='3.0')
- def test_io_is__io(self):
- # _io module calls itself io. This leads
- # to cyclic dependencies when astroid tries to resolve
- # what io.BufferedReader is. The code that handles this
- # is in astroid.raw_building.imported_member, which verifies
- # the true name of the module.
- import _io
-
- builder = AstroidBuilder()
- module = builder.inspect_build(_io)
- buffered_reader = module.getattr('BufferedReader')[0]
- self.assertEqual(buffered_reader.root().name, 'io')
-
- @unittest.skipUnless(os.name == 'java', 'Requires Jython')
- def test_open_is_inferred_correctly(self):
- # Lot of Jython builtins don't have a __module__ attribute.
- for name, _ in inspect.getmembers(builtins, predicate=inspect.isbuiltin):
- if name == 'print':
- continue
- node = extract_node('{0} #@'.format(name))
- inferred = next(node.infer())
- self.assertIsInstance(inferred, nodes.FunctionDef, name)
- self.assertEqual(inferred.root().name, BUILTINS, name)
-
-
- if __name__ == '__main__':
- unittest.main()
|