Spaces:
Sleeping
Sleeping
| import copy | |
| import unittest | |
| from docx import Document | |
| from docx.opc.constants import RELATIONSHIP_TYPE as RT | |
| from docx.oxml import parse_xml | |
| from docx.oxml.ns import qn | |
| from document.evolutivo import construction_characteristics, construction_groups | |
| from test_evo_adjustments import body | |
| from test_evolutivo_documento import fixture, resolved, all_text | |
| def notes(doc): | |
| root = parse_xml(doc.part.part_related_by(RT.FOOTNOTES).blob) | |
| return [''.join(n.itertext(qn('w:t'))) for n in root if int(n.get(qn('w:id'))) > 0] | |
| class ConservationNotesTests(unittest.TestCase): | |
| def data(self): | |
| data = fixture((2021, 2022)) | |
| for year in data['anos']: | |
| year['construcao']['avaliacao']['referenciasCalculo'] = { | |
| 'vidaUtilTipos': [{'tipo': 'GALPOES (DEPOSITOS)', 'vidaUtil': 70}], | |
| 'estadosConservacao': [{'estado': 'C - regular', 'coeficiente': .0252, | |
| 'descricao': 'Descrição histórica da classe regular.'}]} | |
| year['construcao']['construcoes'][0].update(estadoConservacao='C - regular', coefHeidecke=.0317) | |
| return data | |
| def test_codes_and_saved_coefficients_in_tables_descriptions_only_in_notes(self): | |
| imported = resolved(self.data()) | |
| before = copy.deepcopy(imported) | |
| doc = body(imported) | |
| characteristics = next(t for t in doc.tables if t.cell(0, 1).text == 'Tipo / ano') | |
| self.assertEqual([c.text for c in characteristics.rows[0].cells][-2:], ['Estado de conservação', 'H']) | |
| self.assertEqual([c.text for c in characteristics.rows[1].cells][-2:], ['C', '0,0317']) | |
| calculation = next(t for t in doc.tables if any(r.cells[0].text == 'Coeficiente H' for r in t.rows)) | |
| for label, expected in [('Estado de conservação', 'C'), ('Coeficiente H', '0,0317')]: | |
| row = next(r for r in calculation.rows if r.cells[0].text == label) | |
| self.assertEqual([c.text for c in row.cells[1:]], [expected, expected]) | |
| self.assertNotIn('Descrição histórica', all_text(doc)) | |
| self.assertNotIn('Referência de vida útil para', all_text(doc)) | |
| self.assertNotIn('Ajuste manual de vida consumida', all_text(doc)) | |
| footnotes = notes(doc) | |
| self.assertEqual(sum('Descrição histórica da classe regular.' in text for text in footnotes), 1) | |
| self.assertTrue(any('70 anos' in text and 'R é o valor residual' in text for text in footnotes)) | |
| self.assertTrue(any('IPTU 2022; IPTU 2023' in text and 'padrão 50,00%; adotado 40,00%' in text for text in footnotes)) | |
| self.assertNotIn('SEGREDO INTERNO', ' '.join(footnotes)) | |
| self.assertEqual(imported, before) | |
| def test_shared_class_is_explained_once_and_historical_changes_keep_their_scope(self): | |
| data = self.data() | |
| for year in data['anos']: | |
| second = copy.deepcopy(year['construcao']['construcoes'][0]) | |
| second.update(id='segundo', seq=2, nome='Galpão B') | |
| year['construcao']['construcoes'].append(second) | |
| data['anos'][1]['construcao']['avaliacao']['referenciasCalculo']['estadosConservacao'][0]['descricao'] = 'Caracterização revista no segundo exercício.' | |
| doc = Document() | |
| construction_characteristics(doc, construction_groups(resolved(data)['avaliacoes'])) | |
| self.assertEqual(len(doc.tables[0].rows), 5) | |
| for index in (1, 3): | |
| self.assertIn('IPTU 2022', doc.tables[0].cell(index, 0).text) | |
| self.assertIn('IPTU 2023', doc.tables[0].cell(index + 1, 0).text) | |
| self.assertEqual(sum('Descrição histórica' in text for text in notes(doc)), 1) | |
| self.assertEqual(sum('Caracterização revista' in text for text in notes(doc)), 1) | |
| def test_unknown_state_keeps_original_label_without_borrowing_catalog_description(self): | |
| data = self.data() | |
| for year in data['anos']: | |
| year['construcao']['construcoes'][0]['estadoConservacao'] = 'C - classificação especial' | |
| doc = Document() | |
| construction_characteristics(doc, construction_groups(resolved(data)['avaliacoes'])) | |
| self.assertIn('C - classificação especial', ' '.join(notes(doc))) | |
| self.assertNotIn('Descrição histórica', ' '.join(notes(doc))) | |
| self.assertIn('não foi preservada', ' '.join(notes(doc))) | |
| if __name__ == '__main__': | |
| unittest.main() | |