Spaces:
Sleeping
Sleeping
| import copy | |
| import io | |
| import unittest | |
| from unittest.mock import patch | |
| from docx import Document | |
| from docx.oxml.ns import qn | |
| from document.generator import LaudoGenerator | |
| from test_evolutivo_documento import fixture, resolved | |
| class EvolutivoHierarchyTests(unittest.TestCase): | |
| def generate(self, data): | |
| imported = resolved(data) | |
| before = copy.deepcopy(imported) | |
| doc = Document() | |
| with patch('document.evolutivo.matching_local', return_value=(None, '')): | |
| LaudoGenerator()._gerar_corpo(doc, {'avaliacoes_importadas': imported}, {}) | |
| self.assertEqual(imported, before) | |
| stream = io.BytesIO() | |
| doc.save(stream) | |
| return Document(io.BytesIO(stream.getvalue())) | |
| def test_construction_subsections_preserve_parent_numbers_and_word_hierarchy(self): | |
| data = fixture((2021, 2022, 2023, 2024, 2025)) | |
| names = ['Casa principal', 'Edícula de apoio', 'Galpão de ferramentas'] | |
| for year_index, year in enumerate(data['anos']): | |
| original = year['construcao']['construcoes'][0] | |
| year['construcao']['construcoes'] = [] | |
| for index, name in enumerate(names, 1): | |
| c = copy.deepcopy(original) | |
| c.update(id=f'{year_index}-{index}', seq=index, nome=name, | |
| bdiComposicaoAdotada=[{'descricao': 'Administração', 'percentual': 5}]) | |
| year['construcao']['construcoes'].append(c) | |
| doc = self.generate(data) | |
| expected = [f'3.4.1.5.{index} Construção {index} — {name}' for index, name in enumerate(names, 1)] | |
| headings = [p for p in doc.paragraphs if p.text.startswith('3.4.1.5.')] | |
| self.assertEqual([p.text for p in headings], expected) | |
| for p in headings: | |
| self.assertEqual(p._p.pPr.find(qn('w:outlineLvl')).get(qn('w:val')), '4') | |
| self.assertTrue(p.paragraph_format.keep_with_next) | |
| self.assertEqual(p._p.getnext().tag, qn('w:tbl')) | |
| paragraphs = [p.text for p in doc.paragraphs] | |
| bdi_headings = [p.text for p in doc.paragraphs if p.text.startswith('3.4.1.3.')] | |
| self.assertEqual(bdi_headings, [f'3.4.1.3.{index} Construção {index} — {name}' for index, name in enumerate(names, 1)]) | |
| for text in ['3.4.1.1.1 Composição das áreas', | |
| '3.4.1.5 Cálculo por construção e exercício', '3.5 AVALIAÇÃO DO IMÓVEL OBJETO', | |
| '4. ESPECIFICAÇÃO DA AVALIAÇÃO', '5. CONCLUSÃO TÉCNICA']: | |
| self.assertIn(text, paragraphs) | |
| specifications = [p for p in doc.paragraphs if p.text.startswith(('4.1 ', '4.2 ', '4.3 '))] | |
| self.assertEqual(len(specifications), 3) | |
| self.assertTrue(all(p._p.pPr.find(qn('w:outlineLvl')).get(qn('w:val')) == '1' for p in specifications)) | |
| tables = [t for t in doc.tables if any(r.cells[0].text == 'Valor da construção (R$)' for r in t.rows)] | |
| self.assertEqual(len(tables), 3) | |
| self.assertTrue(all(len(t.columns) == 6 for t in tables)) | |
| def test_optional_compositions_do_not_create_empty_subsections(self): | |
| data = fixture((2021,)) | |
| c = data['anos'][0]['construcao']['construcoes'][0] | |
| c.update(areaTotal=100, areaCalculada=100, areasEquivalentes=[{'descricao': 'Área', 'area': 100, 'coeficiente': 1}]) | |
| doc = self.generate(data) | |
| paragraphs = [p.text for p in doc.paragraphs] | |
| self.assertFalse(any(p.endswith(('Composição das áreas', 'Composição do BDI')) for p in paragraphs)) | |
| self.assertEqual([p for p in paragraphs if p.startswith('3.4.1.5.')], ['3.4.1.5.1 Construção 1 — Galpão A']) | |
| self.assertIn('3.5 AVALIAÇÃO DO IMÓVEL OBJETO', paragraphs) | |
| if __name__ == '__main__': | |
| unittest.main() | |