Pequeñas pruebas

pruebas/test_central/test_modelo.py

@@ -154,8 +154,8 @@ def _verificar_gráfico(símismo, **args):
             símismo.res.graficar(dir_, **args)
             símismo.assertTrue(sum([len(files) for r, d, files in os.walk(dir_)]) == 1)
 
-    def test_graficar_res(símismo):
-        símismo._verificar_gráfico()
+    def test_graficar_sin_incert(símismo):
+        símismo._verificar_gráfico(argsll={'incert': None})
 
     def test_graficar_confianza(símismo):
         símismo._verificar_gráfico(argsll={'incert': 'confianza'})

pruebas/test_central/test_modelo_calib.py

@@ -1,6 +1,6 @@
 import unittest
 
-from .rcrs.modelo_calib import modelo, exper, coso
+from .rcrs.modelo_calib import modelo, exper
 
 
 class PruebaCalib(unittest.TestCase):
@@ -12,3 +12,27 @@ def test_calib(símismo):
     @unittest.skip('implementar')
     def test_calib_paráms_mód(símismo):
         pass
+
+    @unittest.skip('implementar')
+    def test_calib_con_inic(símismo):
+        pass
+
+    @unittest.skip('implementar')
+    def test_calib_exluir_inic(símismo):
+        pass
+
+    @unittest.skip('implementar')
+    def test_calib_solamente_inic(símismo):
+        pass
+
+    @unittest.skip('implementar')
+    def test_renombrar_calib(símismo):
+        pass
+
+    @unittest.skip('implementar')
+    def test_borrar_calib(símismo):
+        pass
+
+    @unittest.skip('implementar')
+    def test_guardar_cargar_calib(símismo):
+        pass

pruebas/test_central/test_modelo_funcs_ecs.py

@@ -140,7 +140,8 @@ def test_borrar_apriori(símismo):
         from .rcrs import ec_parám
         exper = ec_parám.exper
         coso = ec_parám.CosoParám('hola')
-        modelo = Modelo(ec_parám.MóduloParám([coso]))
+        módulo = ec_parám.MóduloParám([coso])
+        modelo = Modelo(módulo)
         with símismo.subTest('todos'):
             coso.espec_apriori(
                 apriori=APrioriDist(uniform(3, 1)), categ='categ', sub_categ='subcateg', ec='ec', prm='a'
@@ -165,6 +166,14 @@ def test_borrar_apriori(símismo):
             res = modelo.simular('apriori', exper, t=2)['exper']['módulo']['res']
             símismo._verif_en(res, (3, 3 + 1))
 
+        with símismo.subTest('de modelo'):
+            coso.espec_apriori(
+                apriori=APrioriDist(uniform(3, 1)), categ='categ', sub_categ='subcateg', ec='ec', prm='a'
+            )
+            módulo.borrar_aprioris()
+            res = modelo.simular('apriori', exper, t=2)['exper']['módulo']['res']
+            símismo._verif_fuera_de(res, (3, 3 + 1))
+
     def test_inter(símismo):
         from .rcrs import ec_inter
         modelo = ec_inter.mi_modelo
@@ -196,12 +205,12 @@ def test_inter_aprioris(símismo):
             apriori = APrioriDist(uniform(1.5, 1))
             coso1.espec_apriori(apriori, 'categ', 'subcateg', 'ec', 'a')
             res = modelo.simular('con interacciones', exper=exper, t=2, vars_interés=True)['exper']['módulo']['res']
-            símismo._verif_en(res.datos.loc[{'coso': coso1, 'otro': [coso2, coso3]}], (1.5, 1.5+1))
+            símismo._verif_en(res.datos.loc[{'coso': coso1, 'otro': [coso2, coso3]}], (1.5, 1.5 + 1))
 
         with símismo.subTest('con índs'):
             apriori = APrioriDist(uniform(1.5, 1))
             coso1.borrar_aprioris()
             coso1.espec_apriori(apriori, 'categ', 'subcateg', 'ec', 'a', índs=[coso3])
             res = modelo.simular('con interacciones', exper=exper, t=2, vars_interés=True)['exper']['módulo']['res']
-            símismo._verif_en(res.datos.loc[{'coso': coso1, 'otro': coso3}], (1.5, 1.5+1))
-            símismo._verif_fuera_de(res.datos.loc[{'coso': coso1, 'otro': coso2}], (1.5, 1.5+1))
+            símismo._verif_en(res.datos.loc[{'coso': coso1, 'otro': coso3}], (1.5, 1.5 + 1))
+            símismo._verif_fuera_de(res.datos.loc[{'coso': coso1, 'otro': coso2}], (1.5, 1.5 + 1))

pruebas/test_ecs/test_dists.py

@@ -4,9 +4,10 @@
 import numpy as np
 import numpy.testing as npt
 import scipy.stats as estad
+from matplotlib.figure import Figure as Figura
 from numpy import exp
 from scipy.special import expit
-from tikon.ecs.dists import DistAnalítica, DistTraza, Dist, MnjdrDists
+from tikon.ecs.dists import DistAnalítica, DistTraza, Dist, MnjdrDists, dibujar_dist
 from tikon.ecs.dists.anlt import TransfDist
 from tikon.utils import proc_líms
 
@@ -286,3 +287,10 @@ def test_conv_dic():
         npt.assert_equal(mnjdr.obt_val().aprox_líms(p), nuevo.obt_val().aprox_líms(p))
         npt.assert_equal(mnjdr.obt_val('a').aprox_líms(p), nuevo.obt_val('a').aprox_líms(p))
         npt.assert_equal(mnjdr.obt_val(['a', 'b']).aprox_líms(p), nuevo.obt_val(['a', 'b']).aprox_líms(p))
+
+
+class PruebaDib(unittest.TestCase):
+    def test_dibujar(símismo):
+        dist = DistAnalítica(estad.norm())
+        dib = dibujar_dist(dist, 'Mi distribución')
+        símismo.assertTrue(isinstance(dib.figure, Figura))

tikon/central/módulo.py

@@ -35,7 +35,7 @@ def borrar_aprioris(símismo):
             símismo[c].borrar_aprioris()
 
     def inter(símismo, modelo, coso, tipo):
-        raise ValueError('Interacciones no implementadas para módulo "{mód}".'.format(mód=símismo))
+        raise NotImplementedError('Interacciones no implementadas para módulo "{mód}".'.format(mód=símismo))
 
     @property
     def nombre(símismo):

tikon/central/parc.py

@@ -84,7 +84,7 @@ def _dstn(a, b):
 def _área_de_polígono(polígono):
     proj = partial(
         pyproj.transform,
-        pyproj.Proj(init='epsg:4326'),
+        pyproj.Proj('epsg:4326'),
         pyproj.Proj(
             proj='aea',
             lat_1=polígono.bounds[1],

tikon/ecs/dists/__init__.py

@@ -2,3 +2,4 @@
 from .dists import Dist, MnjdrDists
 from .trz import DistTraza
 from .utils import líms_dist, obt_nombre
+from .dibs import dibujar_dist

tikon/ecs/dibs.py → tikon/ecs/dists/dibs.py

@@ -1,5 +1,8 @@
 import seaborn as sns
 
+from matplotlib.backends.backend_agg import FigureCanvasAgg as TelaFigura
+from matplotlib.figure import Figure as Figura
+
 
 def dibujar_dist(dist, nombre, ejes=None, argsll=None):
     args_base = dict(
@@ -8,6 +11,11 @@ def dibujar_dist(dist, nombre, ejes=None, argsll=None):
     if argsll:
         args_base.update(argsll)
 
+    if ejes is None:
+        fig = Figura()
+        TelaFigura(fig)
+        ejes = fig.add_subplot(111)
+
     puntos = dist.obt_vals(10000)
 
     return sns.kdeplot(puntos, ax=ejes, **args_base).set_title(nombre)

tikon/ecs/árb_coso.py

@@ -57,7 +57,7 @@ def espec_apriori(símismo, apriori, categ, sub_categ, ec, parám, inter=None):
         obj_parám.espec_apriori(apriori, inter=inter)
 
     def activar_ec(símismo, categ, subcateg, ec):
-        símismo[categ][subcateg].activar_ec(ec)
+        símismo[categ].activar_ec(subcateg, ec)
 
     def desactivar_ec(símismo, categ, subcateg=None):
         símismo[categ].desactivar_ec(subcateg=subcateg)