def test_call_module(self):
visitor = doc_generator_visitor.DocGeneratorVisitor()
visitor(
('doc_generator_visitor',), doc_generator_visitor,
[('DocGeneratorVisitor', doc_generator_visitor.DocGeneratorVisitor)])
self.assertEqual({'doc_generator_visitor': ['DocGeneratorVisitor']},
visitor.tree)
self.assertEqual({
'doc_generator_visitor': doc_generator_visitor,
'doc_generator_visitor.DocGeneratorVisitor':
doc_generator_visitor.DocGeneratorVisitor,
}, visitor.index)
def test_call_class(self):
visitor = doc_generator_visitor.DocGeneratorVisitor()
visitor(
('DocGeneratorVisitor',), doc_generator_visitor.DocGeneratorVisitor,
[('index', doc_generator_visitor.DocGeneratorVisitor.reverse_index)])
self.assertEqual({'DocGeneratorVisitor': ['index']},
visitor.tree)
self.assertEqual({
'DocGeneratorVisitor':
doc_generator_visitor.DocGeneratorVisitor,
'DocGeneratorVisitor.index':
doc_generator_visitor.DocGeneratorVisitor.reverse_index
}, visitor.index)
def test_call_class(self):
class ExampleClass:
def example_method(self):
pass
visitor = doc_generator_visitor.DocGeneratorVisitor()
visitor(parent_path=('ExampleClass', ),
parent=ExampleClass,
children=[('example_method', ExampleClass.example_method)])
self.assertEqual({'ExampleClass': ['example_method']}, visitor.tree)
self.assertEqual(
{
'ExampleClass': ExampleClass,
'ExampleClass.example_method': ExampleClass.example_method,
}, visitor.index)
def test_call_raises(self):
visitor = doc_generator_visitor.DocGeneratorVisitor()
with self.assertRaises(RuntimeError):
visitor(('non_class_or_module', ), 'non_class_or_module_object',
[])