def test_guardar_datos_base(self):
from models import DatosBase, Provincia
datos = DatosBase(nombre='ejemplo')
datos.provincia = Provincia.get(id=10)
datos.save()
assert datos.id
assert DatosBase.select().where(nombre="ejemplo").count()
def test_contar_llamadas_por_provincias(self):
from models import Provincia, DatosBase
for p in Provincia.select():
assert p.llamadas.count() == 0
datos = DatosBase(nombre='ejemplo')
datos.provincia = Provincia.get(id=10)
datos.save()
p1 = Provincia.get(id=10)
assert p1.llamadas.count() == 1
def setUp(self):
from models import DatosBase, Provincia
DatosBase.drop_table()
DatosBase.create_table()