def test_get_tree_one_module_one_package(self):
self.assertEqual(
get_tree([
(('root', 'a.py'), (None, None)),
(('root', 'b'), (None, None)),
]),
{
'a.py': None,
'b': {},
}
)
def test_get_tree_module_in_a_package(self):
self.assertEqual(
get_tree([
(('root', 'a'), (None, None)),
(('root', 'a/b.py'), (None, None)),
]),
{
'a': {
'a/b.py': None,
},
}
)
def test_get_tree_nested_packages(self):
self.assertEqual(
get_tree([
(('root', 'a.py'), (None, None)),
(('root', 'b'), (None, None)),
(('root', 'b/c.py'), (None, None)),
(('root', 'b/d'), (None, None)),
(('root', 'b/d/e.py'), (None, None)),
(('root', 'b/f.py'), (None, None)),
(('root', 'g.py'), (None, None)),
]),
{
'a.py': None,
'b': {
'b/c.py': None,
'b/d': {
'b/d/e.py': None,
},
'b/f.py': None,
},
'g.py': None,
}
)