Spaces:
Sleeping
Sleeping
| import copy | |
| import io | |
| import unittest | |
| import zipfile | |
| from lxml import etree | |
| from docx import Document | |
| from docx.oxml import parse_xml | |
| from docx.oxml.ns import qn | |
| from docx.enum.text import WD_ALIGN_PARAGRAPH | |
| from unittest.mock import patch | |
| from document.equations import cost_equations, depreciation_equations | |
| from document.generator import LaudoGenerator | |
| from document.public_text import public_text | |
| from document.footnotes import add_footnote, add_first_footnote | |
| from document.formatters import add_body_text | |
| from test_evolutivo_documento import fixture,resolved | |
| class DepreciationEquationTests(unittest.TestCase): | |
| def test_cost_formulas_preserve_percentage_conversion_and_first_occurrence_note(self): | |
| doc=Document();cost_equations(doc) | |
| formulas=list(doc.element.body.iter(qn('m:oMath'))) | |
| self.assertEqual(len(formulas),2) | |
| self.assertEqual([''.join(t.text or '' for t in m.iter(qn('m:t'))) for m in formulas], | |
| ['CUNIT = CUB × % CUB100','VC = A × CUNIT × (1 + BDI100) × Kd']) | |
| self.assertEqual(len(list(formulas[0].iter(qn('m:f')))),1) | |
| self.assertEqual(len(list(formulas[1].iter(qn('m:f')))),1) | |
| expression_xml=etree.tostring(formulas[0]) | |
| add_first_footnote(doc,r'%\s+CUB','Percentual de adequação do CUB.') | |
| self.assertEqual(etree.tostring(formulas[0]),expression_xml) | |
| self.assertEqual(len(list(doc.paragraphs[0]._p.iter(qn('w:footnoteReference')))),1) | |
| self.assertEqual(len(list(doc.paragraphs[1]._p.iter(qn('w:footnoteReference')))),0) | |
| for index in (1,3): | |
| self.assertEqual({r.font.size.pt for r in doc.paragraphs[index].runs},{8}) | |
| def test_editable_equations_keep_fraction_power_subscript_and_small_legends(self): | |
| doc=Document();depreciation_equations(doc) | |
| stream=io.BytesIO();doc.save(stream);stream.seek(0) | |
| reopened=Document(stream) | |
| math=list(reopened.element.body.iter(qn('m:oMath'))) | |
| self.assertEqual(len(math),3) | |
| self.assertEqual(len(list(math[0].iter(qn('m:f')))),1) | |
| self.assertEqual(len(list(math[0].iter(qn('m:sSup')))),1) | |
| self.assertEqual(len(list(math[2].iter(qn('m:sSub')))),1) | |
| texts=[''.join(t.text or '' for t in formula.iter(qn('m:t'))) for formula in math] | |
| self.assertEqual(texts,['Ross = q + q22','K = 1 − (1 − Ross) × (1 − H)','Kd = R + (1 − K) × (1 − R)']) | |
| self.assertEqual(len(reopened.paragraphs),6) | |
| for index in (0,2,4): | |
| self.assertEqual(reopened.paragraphs[index].alignment,WD_ALIGN_PARAGRAPH.CENTER) | |
| self.assertTrue(reopened.paragraphs[index].paragraph_format.keep_with_next) | |
| legend=reopened.paragraphs[index+1] | |
| self.assertEqual({r.font.size.pt for r in legend.runs},{8}) | |
| self.assertIn('dividindo o percentual',reopened.paragraphs[1].text) | |
| self.assertIn('estado de conservação',reopened.paragraphs[3].text) | |
| self.assertIn('percentual ÷ 100',reopened.paragraphs[5].text) | |
| def test_internal_references_stay_out_of_body_tables_and_footnotes(self): | |
| imported=resolved(fixture((2021,))) | |
| row=imported['avaliacoes'][0] | |
| row['observacoes']='Resultados preservados no MESA.' | |
| row['complementos']={'justificativa_fc':'Condição registrada no modelo MESA.'} | |
| original=copy.deepcopy(imported) | |
| doc=Document() | |
| with patch('document.evolutivo.matching_local',return_value=(None,'')): | |
| LaudoGenerator()._gerar_corpo(doc,{'avaliacoes_importadas':imported},{}) | |
| LaudoGenerator()._gerar_cabecalho(doc,{'modelo_avaliacao':'modelo MESA'}) | |
| add_footnote(doc,add_body_text(doc,'Referência técnica'),'Coeficientes da MESA preservados na avaliação.') | |
| stream=io.BytesIO();doc.save(stream) | |
| with zipfile.ZipFile(stream) as archive: | |
| for name in ['word/document.xml','word/footnotes.xml']: | |
| root=parse_xml(archive.read(name)) | |
| text=' '.join(t.text or '' for t in root.iter(qn('w:t'))) | |
| self.assertNotIn('MESA',text) | |
| self.assertNotIn('versão documental',text) | |
| self.assertNotIn('catálogo sincronizado',text) | |
| self.assertEqual(imported,original) | |
| def test_public_wording_preserves_technical_names_and_ordinary_words(self): | |
| self.assertEqual(public_text('Método de Ross-Heidecke; ABNT NBR 14653-2; CUB.'),'Método de Ross-Heidecke; ABNT NBR 14653-2; CUB.') | |
| self.assertEqual(public_text('modelo MESA'),'modelo estatístico') | |
| self.assertEqual(public_text('salvo no MESA'),'salvo na avaliação de origem') | |
| self.assertEqual(public_text('mesa de apoio'),'mesa de apoio') | |
| if __name__=='__main__': | |
| unittest.main() | |