def test_toc_with_subparts(self):
c1 = Node(label=['205', '2'], title='Authority and Purpose')
c2 = Node(label=['205', '3'], title='Definitions')
c3 = Node(label=['205', '4'], title='Coverage')
s1 = Node(children=[c1, c2, c3],
label=['205', 'Subpart', 'A'],
title='First Subpart',
node_type=Node.SUBPART)
c4 = Node(label=['205', '5'], title='Fifth Title')
s2 = Node(children=[c4],
label=['205', 'Subpart', 'B'],
title='Second Subpart',
node_type=Node.SUBPART)
n = Node(children=[s1, s2], label=['205'])
parser = table_of_contents.TableOfContentsLayer(None)
toc = parser.process(n)
self.assertEqual(len(toc), 2)
self.assertEqual(toc[0]['index'], ['205', 'Subpart', 'A'])
self.assertEqual(toc[1]['index'], ['205', 'Subpart', 'B'])
toc = parser.process(s1)
self.assertEqual(len(toc), 3)
self.assertEqual(toc[0]['index'], ['205', '2'])
self.assertEqual(toc[1]['index'], ['205', '3'])
self.assertEqual(toc[2]['index'], ['205', '4'])
toc = parser.process(s2)
self.assertEqual(len(toc), 1)
self.assertEqual(toc[0]['index'], ['205', '5'])
def test_toc_generation(self):
c1 = Node(label=['1005', '2'], title='Authority and Purpose')
c2 = Node(label=['1005', '3'], title='Definitions')
c3 = Node(label=['1005', '4'], title='Coverage')
n = Node(children=[c1, c2, c3], label=['1005'])
parser = table_of_contents.TableOfContentsLayer(None)
toc = parser.process(n)
self.assertEqual(len(toc), 3)
self.assertEqual(toc[0].keys(), ['index', 'title'])
self.assertEqual(toc[0]['index'], ['1005', '2'])
self.assertEqual(toc[1]['index'], ['1005', '3'])
self.assertEqual(toc[2]['index'], ['1005', '4'])
def test_build_tree(self):
c1 = Node(label=['1005', 'A', '2'], title='Authority and Purpose')
c2 = Node(label=['1005', 'A', '3'], title='Definitions')
c3 = Node(label=['1005', 'A', '4'], title='Coverage')
a1 = Node(children=[c1, c2, c3], label=['1005', 'A'])
tree = Node(children=[a1], label=['1005'], title='Zombie Manifesto')
parser = table_of_contents.TableOfContentsLayer(tree)
toc_layer = parser.build()
self.assertEquals(toc_layer.keys(), ['1005-A'])
self.assertEqual(len(toc_layer['1005-A']), 3)
self.assertEqual(toc_layer['1005-A'][0]['index'], ['1005', 'A', '2'])
def test_toc_with_emptysubpart(self):
p1 = Node(label=['205', '2'], title='Authority and Purpose')
p2 = Node(label=['205', '3'], title='Definitions')
p3 = Node(label=['205', '4'], title='Coverage')
s1 = Node(children=[p1, p2, p3],
label=['205', 'Subpart'],
node_type=Node.EMPTYPART)
a1 = Node(label=['205', 'A'], title='Appendix A')
interp = Node(label=['205', 'Interp'], title='Supplement I')
n = Node(children=[s1, a1, interp], label=['205'])
parser = table_of_contents.TableOfContentsLayer(None)
toc = parser.process(n)
self.assertEqual(len(toc), 5)
def generate_table_of_contents(reg_json):
""" Generate the Table of Contents layer """
layer_generator = table_of_contents.TableOfContentsLayer(reg_json)
toc = layer_generator.build()
print NodeEncoder().encode(toc)
def test_toc_no_children(self):
tree = Node(label=['1005'])
parser = table_of_contents.TableOfContentsLayer(None)
toc = parser.process(tree)
self.assertEqual(len(toc), 0)