bee_python 0.0.1

data/tools/logilab/logilab/astng/test/unittest_inspector.py

@@ -0,0 +1,105 @@
+# Copyright (c) 2003-2006 LOGILAB S.A. (Paris, FRANCE).
+# http://www.logilab.fr/ -- mailto:contact@logilab.fr
+#
+# This program is free software; you can redistribute it and/or modify it under
+# the terms of the GNU Lesser General Public License as published by the Free Software
+# Foundation; either version 2 of the License, or (at your option) any later
+# version.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License along with
+# this program; if not, write to the Free Software Foundation, Inc.,
+# 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+# copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+# copyright 2003-2010 Sylvain Thenault, all rights reserved.
+# contact mailto:thenault@gmail.com
+#
+# This file is part of logilab-astng.
+#
+# logilab-astng is free software: you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as published by the
+# Free Software Foundation, either version 2.1 of the License, or (at your
+# option) any later version.
+#
+# logilab-astng is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
+# for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License along
+# with logilab-astng. If not, see <http://www.gnu.org/licenses/>.
+"""
+unittest for the visitors.diadefs module
+"""
+
+import unittest
+import sys
+
+from logilab.astng import MANAGER, nodes, inspector
+from logilab.astng.bases import Instance, YES
+
+def astng_wrapper(func, modname):
+    return func(modname)
+
+        
+class LinkerTC(unittest.TestCase):
+    
+    def setUp(self):
+        self.project = MANAGER.project_from_files(['data2'], astng_wrapper) 
+        self.linker = inspector.Linker(self.project)
+        self.linker.visit(self.project)
+
+    def test_class_implements(self):
+        klass = self.project.get_module('data2.clientmodule_test')['Ancestor']
+        self.assert_(hasattr(klass, 'implements'))
+        self.assertEqual(len(klass.implements), 1)
+        self.assert_(isinstance(klass.implements[0], nodes.Class))
+        self.assertEqual(klass.implements[0].name, "Interface")
+        klass = self.project.get_module('data2.clientmodule_test')['Specialization']
+        self.assert_(hasattr(klass, 'implements'))
+        self.assertEqual(len(klass.implements), 0)
+        
+    def test_locals_assignment_resolution(self):
+        klass = self.project.get_module('data2.clientmodule_test')['Specialization']
+        self.assert_(hasattr(klass, 'locals_type'))
+        type_dict = klass.locals_type
+        self.assertEqual(len(type_dict), 2)
+        keys = type_dict.keys()
+        keys.sort()
+        self.assertEqual(keys, ['TYPE', 'top'])
+        self.assertEqual(len(type_dict['TYPE']), 1)
+        self.assertEqual(type_dict['TYPE'][0].value, 'final class')
+        self.assertEqual(len(type_dict['top']), 1)
+        self.assertEqual(type_dict['top'][0].value, 'class')
+        
+    def test_instance_attrs_resolution(self):
+        klass = self.project.get_module('data2.clientmodule_test')['Specialization']
+        self.assert_(hasattr(klass, 'instance_attrs_type'))
+        type_dict = klass.instance_attrs_type
+        self.assertEqual(len(type_dict), 3)
+        keys = type_dict.keys()
+        keys.sort()
+        self.assertEqual(keys, ['_id', 'relation', 'toto'])
+        self.assert_(isinstance(type_dict['relation'][0], Instance), type_dict['relation'])
+        self.assertEqual(type_dict['relation'][0].name, 'DoNothing')
+        self.assert_(isinstance(type_dict['toto'][0], Instance), type_dict['toto'])
+        self.assertEqual(type_dict['toto'][0].name, 'Toto')
+        self.assert_(type_dict['_id'][0] is YES, type_dict['_id'])
+
+
+class LinkerTC2(LinkerTC):
+    
+    def setUp(self):
+        self.project = MANAGER.from_directory('data2') 
+        self.linker = inspector.Linker(self.project)
+        self.linker.visit(self.project)
+        
+__all__ = ('LinkerTC', 'LinkerTC2')
+
+        
+if __name__ == '__main__':
+    unittest.main()

data/tools/logilab/logilab/astng/test/unittest_lookup.py

@@ -0,0 +1,302 @@
+# This program is free software; you can redistribute it and/or modify it under
+# the terms of the GNU Lesser General Public License as published by the Free Software
+# Foundation; either version 2 of the License, or (at your option) any later
+# version.
+# copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+# copyright 2003-2010 Sylvain Thenault, all rights reserved.
+# contact mailto:thenault@gmail.com
+#
+# This file is part of logilab-astng.
+#
+# logilab-astng is free software: you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as published by the
+# Free Software Foundation, either version 2.1 of the License, or (at your
+# option) any later version.
+#
+# logilab-astng is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
+# for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License along
+# with logilab-astng. If not, see <http://www.gnu.org/licenses/>.
+
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
+
+# You should have received a copy of the GNU Lesser General Public License along with
+# this program; if not, write to the Free Software Foundation, Inc.,
+# 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+"""tests for the astng variable lookup capabilities
+"""
+import sys
+from os.path import join, abspath
+from logilab.common.testlib import TestCase, unittest_main
+
+from logilab.astng import builder, nodes, scoped_nodes, \
+     InferenceError, NotFoundError
+from logilab.astng.scoped_nodes import builtin_lookup
+from unittest_inference import get_name_node
+
+builder = builder.ASTNGBuilder()
+MODULE = builder.file_build('data/module.py', 'data.module')
+MODULE2 = builder.file_build('data/module2.py', 'data.module2')
+NONREGR = builder.file_build('data/nonregr.py', 'data.nonregr')
+
+class LookupTC(TestCase):
+
+    def test_limit(self):
+        code = '''
+l = [a
+     for a,b in list]
+
+a = 1
+b = a
+a = None
+
+def func():
+    c = 1
+        '''
+        astng = builder.string_build(code, __name__, __file__)
+        names = astng.nodes_of_class(nodes.Name)
+        a = names.next()
+        stmts = a.lookup('a')[1]
+        self.failUnlessEqual(len(stmts), 1)
+        b = astng.locals['b'][1]
+        #self.failUnlessEqual(len(b.lookup('b')[1]), 1)
+        self.failUnlessEqual(len(astng.lookup('b')[1]), 2)
+        b_infer = b.infer()
+        b_value = b_infer.next()
+        self.failUnlessEqual(b_value.value, 1)
+        self.failUnlessRaises(StopIteration, b_infer.next)
+        func = astng.locals['func'][0]
+        self.failUnlessEqual(len(func.lookup('c')[1]), 1)
+
+    def test_module(self):
+        astng = builder.string_build('pass', __name__, __file__)
+        # built-in objects
+        none = astng.ilookup('None').next()
+        self.assertEquals(none.value, None)
+        obj = astng.ilookup('object').next()
+        self.assertIsInstance(obj, nodes.Class)
+        self.assertEquals(obj.name, 'object')
+        self.assertRaises(InferenceError, astng.ilookup('YOAA').next)
+
+        # XXX
+        self.assertEquals(len(list(NONREGR.ilookup('enumerate'))), 2)
+
+    def test_class_ancestor_name(self):
+        code = '''
+class A:
+    pass
+
+class A(A):
+    pass
+        '''
+        astng = builder.string_build(code, __name__, __file__)
+        cls1 = astng.locals['A'][0]
+        cls2 = astng.locals['A'][1]
+        name = cls2.nodes_of_class(nodes.Name).next()
+        self.assertEquals(name.infer().next(), cls1)
+        
+    ### backport those test to inline code
+    def test_method(self):
+        method = MODULE['YOUPI']['method']
+        my_dict = method.ilookup('MY_DICT').next()
+        self.assert_(isinstance(my_dict, nodes.Dict), my_dict)
+        none = method.ilookup('None').next()
+        self.assertEquals(none.value, None)
+        self.assertRaises(InferenceError, method.ilookup('YOAA').next)
+
+        
+    def test_function_argument_with_default(self):
+        make_class = MODULE2['make_class']
+        base = make_class.ilookup('base').next()
+        self.assert_(isinstance(base, nodes.Class), base.__class__)
+        self.assertEquals(base.name, 'YO')
+        self.assertEquals(base.root().name, 'data.module')
+
+
+    def test_class(self):
+        klass = MODULE['YOUPI']
+        my_dict = klass.ilookup('MY_DICT').next()
+        self.assertIsInstance(my_dict, nodes.Dict)
+        none = klass.ilookup('None').next()
+        self.assertEquals(none.value, None)
+        obj = klass.ilookup('object').next()
+        self.assertIsInstance(obj, nodes.Class)
+        self.assertEquals(obj.name, 'object')
+        self.assertRaises(InferenceError, klass.ilookup('YOAA').next)
+
+
+    def test_inner_classes(self):
+        ccc = NONREGR['Ccc']
+        self.assertEquals(ccc.ilookup('Ddd').next().name, 'Ddd')
+
+
+    def test_loopvar_hiding(self):
+        astng = builder.string_build("""
+x = 10
+for x in range(5):
+    print x
+   
+if x > 0:
+    print '#' * x        
+        """, __name__, __file__)
+        xnames = [n for n in astng.nodes_of_class(nodes.Name) if n.name == 'x']
+        # inside the loop, only one possible assignment
+        self.assertEquals(len(xnames[0].lookup('x')[1]), 1)
+        # outside the loop, two possible assignments
+        self.assertEquals(len(xnames[1].lookup('x')[1]), 2)
+        self.assertEquals(len(xnames[2].lookup('x')[1]), 2)
+
+    def test_list_comps(self):
+        if sys.version_info < (2, 4):
+            self.skip('this test require python >= 2.4')
+        astng = builder.string_build("""
+print [ i for i in range(10) ]
+print [ i for i in range(10) ]
+print list( i for i in range(10) )
+        """, __name__, __file__)
+        xnames = [n for n in astng.nodes_of_class(nodes.Name) if n.name == 'i']
+        self.assertEquals(len(xnames[0].lookup('i')[1]), 1)
+        self.assertEquals(xnames[0].lookup('i')[1][0].lineno, 2)
+        self.assertEquals(len(xnames[1].lookup('i')[1]), 1)
+        self.assertEquals(xnames[1].lookup('i')[1][0].lineno, 3)
+        self.assertEquals(len(xnames[2].lookup('i')[1]), 1)
+        self.assertEquals(xnames[2].lookup('i')[1][0].lineno, 4)
+
+
+    def test_explicit___name__(self):
+        code = '''
+class Pouet:
+    __name__ = "pouet"
+p1 = Pouet()
+
+class PouetPouet(Pouet): pass
+p2 = Pouet()
+
+class NoName: pass
+p3 = NoName()
+'''
+        astng = builder.string_build(code, __name__, __file__)
+        p1 = astng['p1'].infer().next()
+        self.failUnless(p1.getattr('__name__'))
+        p2 = astng['p2'].infer().next()
+        self.failUnless(p2.getattr('__name__'))
+        self.failUnless(astng['NoName'].getattr('__name__'))
+        p3 = astng['p3'].infer().next()
+        self.assertRaises(NotFoundError, p3.getattr, '__name__')
+
+
+    def test_function_module_special(self):
+        astng = builder.string_build('''
+def initialize(linter):
+    """initialize linter with checkers in this package """
+    package_load(linter, __path__[0])
+        ''', 'data.__init__', 'data/__init__.py')
+        path = [n for n in astng.nodes_of_class(nodes.Name) if n.name == '__path__'][0]
+        self.assertEquals(len(path.lookup('__path__')[1]), 1)
+
+
+    def test_builtin_lookup(self):
+        self.assertEquals(builtin_lookup('__dict__')[1], ())
+        intstmts = builtin_lookup('int')[1]
+        self.assertEquals(len(intstmts), 1)
+        self.assertIsInstance(intstmts[0], nodes.Class)
+        self.assertEquals(intstmts[0].name, 'int')
+        self.assertIs(intstmts[0], nodes.const_factory(1)._proxied)
+
+
+    def test_decorator_arguments_lookup(self):
+        if sys.version_info < (2, 4):
+            self.skip('this test require python >= 2.4')
+        code = '''
+def decorator(value):
+   def wrapper(function):
+        return function
+   return wrapper
+
+class foo:
+  member = 10
+
+  @decorator(member) #This will cause pylint to complain
+  def test(self):
+       pass
+        ''' 
+        astng = builder.string_build(code, __name__, __file__)
+        member = get_name_node(astng['foo'], 'member')
+        it = member.infer()
+        obj = it.next()
+        self.assertIsInstance(obj, nodes.Const)
+        self.assertEquals(obj.value, 10)
+        self.assertRaises(StopIteration, it.next)
+
+       
+    def test_inner_decorator_member_lookup(self):
+        if sys.version_info < (2, 4):
+            self.skip('this test require python >= 2.4')
+        code = '''
+class FileA:
+    def decorator(bla):
+        return bla
+    
+    @decorator
+    def funcA():
+        return 4
+        '''
+        astng = builder.string_build(code, __name__, __file__)
+        decname = get_name_node(astng['FileA'], 'decorator')
+        it = decname.infer()
+        obj = it.next()
+        self.assertIsInstance(obj, nodes.Function)
+        self.assertRaises(StopIteration, it.next)
+        
+        
+    def test_static_method_lookup(self):
+        if sys.version_info < (2, 4):
+            self.skip('this test require python >= 2.4')
+        code = '''
+class FileA:
+    @staticmethod
+    def funcA():
+        return 4
+
+
+class Test:
+    FileA = [1,2,3]
+    
+    def __init__(self):
+        print FileA.funcA()
+        '''
+        astng = builder.string_build(code, __name__, __file__)
+        it = astng['Test']['__init__'].ilookup('FileA')
+        obj = it.next()
+        self.assertIsInstance(obj, nodes.Class)
+        self.assertRaises(StopIteration, it.next)
+
+
+    def test_global_delete(self):
+        code = '''
+def run2():
+    f = Frobble()
+    
+class Frobble:
+     pass
+Frobble.mumble = True
+
+del Frobble
+
+def run1():
+    f = Frobble()
+'''
+        astng = builder.string_build(code, __name__, __file__)
+        stmts = astng['run2'].lookup('Frobbel')[1]
+        self.failUnlessEqual(len(stmts), 0)
+        stmts = astng['run1'].lookup('Frobbel')[1]
+        self.failUnlessEqual(len(stmts), 0)
+        
+if __name__ == '__main__':
+    unittest_main()

data/tools/logilab/logilab/astng/test/unittest_manager.py

@@ -0,0 +1,98 @@
+# copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+# copyright 2003-2010 Sylvain Thenault, all rights reserved.
+# contact mailto:thenault@gmail.com
+#
+# This file is part of logilab-astng.
+#
+# logilab-astng is free software: you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as published by the
+# Free Software Foundation, either version 2.1 of the License, or (at your
+# option) any later version.
+#
+# logilab-astng is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
+# for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License along
+# with logilab-astng. If not, see <http://www.gnu.org/licenses/>.
+import unittest
+import os
+import sys
+from os.path import join, dirname
+from logilab.astng.manager import ASTNGManager
+
+
+class ASTNGManagerTC(unittest.TestCase):
+    def setUp(self):
+        self.manager = ASTNGManager(borg=False)
+        
+    def test_astng_from_module(self):
+        astng = self.manager.astng_from_module(unittest)
+        self.assertEquals(astng.pure_python, True)
+        import time
+        astng = self.manager.astng_from_module(time)
+        self.assertEquals(astng.pure_python, False)
+        
+    def test_astng_from_class(self):
+        astng = self.manager.astng_from_class(file)
+        self.assertEquals(astng.name, 'file')
+        self.assertEquals(astng.parent.frame().name, '__builtin__')
+
+        astng = self.manager.astng_from_class(object)
+        self.assertEquals(astng.name, 'object')
+        self.assertEquals(astng.parent.frame().name, '__builtin__')
+        self.failUnless('__setattr__' in astng)
+        
+    def _test_astng_from_zip(self, archive):
+        origpath = sys.path[:]
+        sys.modules.pop('mypypa', None)
+        sys.path.insert(0, join(dirname(__file__), 'data', archive))
+        try:
+            module = self.manager.astng_from_module_name('mypypa')
+            self.assertEquals(module.name, 'mypypa')
+            self.failUnless(module.file.endswith('%s/mypypa' % archive),
+                            module.file)
+        finally:
+            sys.path = origpath
+
+    def test_astng_from_module_name_egg(self):
+        self._test_astng_from_zip('MyPyPa-0.1.0-py2.5.egg')
+
+    def test_astng_from_module_name_zip(self):
+        self._test_astng_from_zip('MyPyPa-0.1.0-py2.5.zip')            
+        
+    def test_from_directory(self):
+        obj = self.manager.from_directory('data')
+        self.assertEquals(obj.name, 'data')
+        self.assertEquals(obj.path, join(os.getcwd(), 'data'))
+        
+    def test_package_node(self):
+        obj = self.manager.from_directory('data')
+        expected_short = ['SSL1', '__init__', 'all', 'appl', 'format', 'module', 'module2',
+                          'noendingnewline', 'nonregr', 'notall']
+        expected_long = ['SSL1', 'data', 'data.all', 'appl', 'data.format', 'data.module',
+                         'data.module2', 'data.noendingnewline', 'data.nonregr',
+                         'data.notall']
+        self.assertEquals(obj.keys(), expected_short)
+        self.assertEquals([m.name for m in obj.values()], expected_long)
+        self.assertEquals([m for m in list(obj)], expected_short)
+        self.assertEquals([(name, m.name) for name, m in obj.items()],
+                          zip(expected_short, expected_long))
+        self.assertEquals([(name, m.name) for name, m in obj.items()],
+                          zip(expected_short, expected_long))
+        
+        self.assertEquals('module' in obj, True)
+        self.assertEquals(obj.has_key('module'), True)
+        self.assertEquals(obj.get('module').name, 'data.module')
+        self.assertEquals(obj['module'].name, 'data.module')
+        self.assertEquals(obj.get('whatever'), None)
+        self.assertEquals(obj.fullname(), 'data')
+        # FIXME: test fullname on a subpackage
+
+        
+if __name__ == '__main__':
+    unittest.main()
+
+