def testHierarchy(self):
class MockNode(object):
node = 1
node = MockNode()
codeFinder = CodeFinder()
codeFinder.visit = lambda _: None
codeFinder.package = 'TestPackage'
codeFinder.module = '__init__'
codeFinder.visitModule(node)
codeFinder.module = 'Modulo'
codeFinder.visitModule(node)
codeFinder.package = 'TestPackage.Another'
codeFinder.module = '__init__'
codeFinder.visitModule(node)
codeFinder.module = 'Moduli'
codeFinder.visitModule(node)
expected = [
'TestPackage',
'TestPackage.Modulo',
'TestPackage.Another',
'TestPackage.Another.Moduli',
]
self.assertEquals(codeFinder.modules['HIERARCHY'], expected)
def testHierarchy(self):
class MockNode(object):
node = 1
node = MockNode()
codeFinder = CodeFinder()
codeFinder.visit = lambda _: None
codeFinder.package = 'TestPackage'
codeFinder.module = '__init__'
codeFinder.visitModule(node)
codeFinder.module = 'Modulo'
codeFinder.visitModule(node)
codeFinder.package = 'TestPackage.Another'
codeFinder.module = '__init__'
codeFinder.visitModule(node)
codeFinder.module = 'Moduli'
codeFinder.visitModule(node)
expected = [
'TestPackage',
'TestPackage.Modulo',
'TestPackage.Another',
'TestPackage.Another.Moduli',
]
self.assertEqual(codeFinder.modules['HIERARCHY'], expected)
def testOnlyPackage(self):
source = """
class A(object):
pass
"""
tree = compiler.parse(dedent(source))
codeFinder = CodeFinder()
codeFinder.package = 'TestPackage'
codeFinder.module = '__init__'
compiler.walk(tree, codeFinder)
expected = {
'CLASSES': {
'TestPackage.A':
dict(docstring='',
bases=['object'],
constructor=[],
methods=[],
properties=[])
},
'FUNCTIONS': [],
'CONSTANTS': [],
'POINTERS': {},
'HIERARCHY': ['TestPackage']
}
actual = eval(pformat(codeFinder.modules))
self.assertEquals(actual, expected)
def getModule(self, source):
tree = compiler.parse(dedent(source))
codeFinder = CodeFinder()
codeFinder.module = 'TestModule'
codeFinder.package = 'TestPackage'
compiler.walk(tree, codeFinder, walker=ExampleASTVisitor(), verbose=1)
try:
return eval(pformat(codeFinder.modules))
except:
print 'EXCEPTION WHEN EVALING:'
print pformat(codeFinder.modules)
print '=-' * 20
raise
def getModule(self, source):
tree = compiler.parse(dedent(source))
codeFinder = CodeFinder()
codeFinder.module = 'TestModule'
codeFinder.package = 'TestPackage'
compiler.walk(tree, codeFinder)
try:
return eval(pformat(codeFinder.modules))
except:
print('EXCEPTION WHEN EVALING:')
print(pformat(codeFinder.modules))
print('=-' * 20)
raise
def testOnlyPackage(self):
source = """
class A(object):
pass
"""
tree = compiler.parse(dedent(source))
codeFinder = CodeFinder()
codeFinder.package = 'TestPackage'
codeFinder.module = '__init__'
compiler.walk(tree, codeFinder, walker=ExampleASTVisitor(), verbose=1)
expected = {'CLASSES': {'TestPackage.A': dict(docstring='', bases=['object'], constructor=[], methods=[], properties=[])},
'FUNCTIONS': [], 'CONSTANTS': [], 'POINTERS': {}, 'HIERARCHY': ['TestPackage']}
actual = eval(pformat(codeFinder.modules))
self.assertEquals(actual, expected)
