run_tests.py

#!/usr/bin/env python

import os
from lxml import etree
from calc import Calculator
from val import Validator
from data import *
from ref_data import RefData

parser = etree.XMLParser(remove_comments=True, encoding='utf8')

xml = etree.parse(os.path.join('tests','104ORI_MAR2011.xml'), parser)

def test_calcs(version):
    print 'Prueba de calculos:'
    calcs = Calculator()
    calcs.load_xml(version)
    calcs.load_xsl('calculos.xsl')

    calcs.calc(xml)

    for x in calcs.get_calculations():
        print x
    print 'Listo'


def test_vals(version):
    print 'Prueba de validaciones:'
    valid = Validator()
    valid.load_xml(version)
    valid.load_xsl('validaciones.xsl')

    valid.validate(xml)

    for x in valid.get_validations():
        print x
    print 'Listo'


def test_xml_file_load():
    print 'Prueba de carga:'
    ref_data = RefData()
        
    lstContribuyentes = ListaContribuyentes()
    lstContribuyentes.load()
        
    declaracion = Declaracion()
    try:
        declaracion = declaracion.cargar_declaracion_guardada(xml, lstContribuyentes, ref_data)
        print 'Listo'
    except Exception as ex:
        print str(ex)
        return 

    test_calcs(declaracion.get_version())

    test_vals(declaracion.get_version())



def run_tests():
    test_xml_file_load()


if __name__ == '__main__':
    run_tests()