diff --git a/sapl/base/tests/test_form.py b/sapl/base/tests/test_form.py index 35f1c82bf..03362fdcd 100644 --- a/sapl/base/tests/test_form.py +++ b/sapl/base/tests/test_form.py @@ -1,5 +1,6 @@ import pytest from django.utils.translation import ugettext_lazy as _ +from model_mommy import mommy from sapl.base.forms import CasaLegislativaForm diff --git a/sapl/sessao/tests/test_sessao_view.py b/sapl/sessao/tests/test_sessao_view.py index 082652c19..b64daefbd 100644 --- a/sapl/sessao/tests/test_sessao_view.py +++ b/sapl/sessao/tests/test_sessao_view.py @@ -3,11 +3,11 @@ from django.core.urlresolvers import reverse from django.utils.translation import ugettext_lazy as _ from model_mommy import mommy -from sapl.parlamentares.models import Legislatura, SessaoLegislativa +from sapl.parlamentares.models import Legislatura, Partido, SessaoLegislativa +from sapl.sessao import forms from sapl.sessao.models import SessaoPlenaria, TipoSessaoPlenaria - -@pytest.mark.django_db(transaction=False) +pytest.mark.django_db(transaction=False) def test_incluir_sessao_plenaria_submit(admin_client): legislatura = mommy.make(Legislatura) sessao = mommy.make(SessaoLegislativa)