def test_nested_sublist_parent_contains_child_and_child_parent(self):
components = [
self.create_numbering_paragraph('1', '0'),
self.create_numbering_paragraph('2', '1'),
self.create_numbering_paragraph('2', '1'),
self.create_numbering_paragraph('1', '0'),
self.create_numbering_paragraph('2', '1'),
self.create_numbering_paragraph('2', '2'),
]
builder = NumberingSpanBuilder(components)
result = builder.detect_parent_child_map_for_items()
parent_items = {
('1', '0'): [
{
'num_id': '2',
'level': '1'
},
]
}
child_item = {
'2': {
'num_id': '1',
'level': '0'
},
}
self.assertDictEqual(builder.parent_child_num_map, parent_items)
self.assertDictEqual(builder.child_parent_num_map, child_item)
self.assertTrue(result)
def test_valid_input_components_but_no_sublists_found(self):
components = [
Paragraph(),
self.create_numbering_paragraph('1', '0'),
self.create_numbering_paragraph('1', '0'),
self.create_numbering_paragraph('2', '0'),
self.create_numbering_paragraph('2', '0'),
self.create_numbering_paragraph('3', '0'),
self.create_numbering_paragraph('3', '0'),
]
list_start_stop_index = {
'1': {
'start': 1,
'stop': 2
},
'2': {
'start': 3,
'stop': 4
},
'3': {
'start': 5,
'stop': 6
},
}
builder = NumberingSpanBuilder(components)
result = builder.detect_parent_child_map_for_items()
self.assertDictEqual(builder.parent_child_num_map, {})
self.assertDictEqual(builder.child_parent_num_map, {})
self.assertDictEqual(builder.list_start_stop_index,
list_start_stop_index)
self.assertTrue(result)
def test_no_components_on_init(self):
builder = NumberingSpanBuilder()
result = builder.detect_parent_child_map_for_items()
self.assertEqual(builder.child_parent_num_map, {})
self.assertEqual(builder.parent_child_num_map, {})
self.assertFalse(result)
def test_nested_sublist_not_wrapped_in_parent_item(self):
components = [
self.create_numbering_paragraph('1', '0'),
self.create_numbering_paragraph('2', '0'),
self.create_numbering_paragraph('3', '0'),
self.create_numbering_paragraph('3', '0'),
self.create_numbering_paragraph('1', '0'),
]
builder = NumberingSpanBuilder(components)
result = builder.detect_parent_child_map_for_items()
parent_items = {
('1', '0'): [
{
'num_id': '2',
'level': '0'
},
{
'num_id': '3',
'level': '0'
},
]
}
child_item = {
'2': {
'num_id': '1',
'level': '0'
},
'3': {
'num_id': '1',
'level': '0'
},
}
list_start_stop_index = {
'1': {
'start': 0,
'stop': 4
},
'2': {
'start': 1,
'stop': 1
},
'3': {
'start': 2,
'stop': 3
},
}
self.assertDictEqual(builder.parent_child_num_map, parent_items)
self.assertDictEqual(builder.child_parent_num_map, child_item)
self.assertDictEqual(builder.list_start_stop_index,
list_start_stop_index)
self.assertTrue(result)
def test_invalid_input_components(self):
components = [
Paragraph(),
Paragraph(children=[Run(children=[
TabChar(),
])]),
Paragraph(properties=ParagraphProperties()),
Paragraph(properties=ParagraphProperties(
numbering_properties=NumberingProperties())),
self.create_numbering_paragraph('1', '0', container=False),
]
builder = NumberingSpanBuilder(components)
result = builder.detect_parent_child_map_for_items()
self.assertEqual(builder.parent_child_num_map, {})
self.assertEqual(builder.child_parent_num_map, {})
self.assertFalse(result)
def setUp(self):
self.builder = NumberingSpanBuilder()