Tests passing

2 files changed, 5 insertions, 6 deletions

diff --git a/engagements/tests/test_forms.py b/engagements/tests/test_forms.py
index 3c213c5..b9807dd 100644
--- a/engagements/tests/test_forms.py
+++ b/engagements/tests/test_forms.py
@@ -1,8 +1,7 @@
 import pytest
 from django.forms import DateInput, Select, SelectMultiple
 
-from engagements.forms import EngagementCreateForm
-from engagements.forms import EngagementEffortCreateForm
+from engagements.forms import EngagementCreateForm, EngagementEffortCreateForm
 from engagements.models import EngagementType
 
 pytestmark = pytest.mark.django_db
@@ -102,10 +101,10 @@ def test_form_missing_required_fields():
     assert "external_party" in form.errors
 
 
-def test_form_engagement_type_queryset():
+def test_form_engagement_type_queryset(engagement_type):
     initial_data = {"engagement_type": EngagementType.objects.all()}
     form = EngagementCreateForm(initial=initial_data)
-    assert form.fields["engagement_type"].queryset == initial_data["engagement_type"]
+    assert form.fields["engagement_type"].queryset.first().name == initial_data["engagement_type"].first().name
 
 
 def test_form_engagement_type_queryset_without_initial():
diff --git a/engagements/views.py b/engagements/views.py
index 88f2014..e66d679 100644
--- a/engagements/views.py
+++ b/engagements/views.py
@@ -2,12 +2,12 @@ from django.contrib.auth.decorators import login_required
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.db.models import Q
 from django.http import HttpResponse
-from django.shortcuts import get_object_or_404
-from django.shortcuts import redirect, render
+from django.shortcuts import get_object_or_404, redirect, render
 from django.template.loader import render_to_string
 from django.views.generic import ListView
 
 from instruments.models import SubInstrument
+
 from .forms import (
     EngagementCreateForm,
     EngagementEffortPlanningCreateForm,