Skip to content

Commit 7fbc81f

Browse files
committed
Fix unit test
1 parent 85a6a9a commit 7fbc81f

File tree

1 file changed

+13
-12
lines changed

1 file changed

+13
-12
lines changed

tests/model_fields_/test_embedded_model_array.py

Lines changed: 13 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def setUpTestData(cls):
122122
exhibit_name="New Discoveries",
123123
sections=[
124124
ExhibitSection(
125-
section_number=1,
125+
section_number=2,
126126
artifacts=[
127127
ArtifactDetail(
128128
name="Lighthouse of Alexandria",
@@ -167,13 +167,13 @@ def test_filter_with_field(self):
167167
def test_filter_with_embeddedfield_path(self):
168168
self.assertCountEqual(
169169
MuseumExhibit.objects.filter(sections__0__section_number=1),
170-
[self.egypt, self.wonders, self.new_descoveries],
170+
[self.egypt, self.wonders],
171171
)
172172

173173
def test_filter_with_embeddedfield_array_path(self):
174174
self.assertCountEqual(
175175
MuseumExhibit.objects.filter(
176-
main_section__artifacts__restorations__0__restored_by="Restoration Lab A"
176+
main_section__artifacts__restorations__0__restored_by="Zacarias"
177177
),
178178
[self.lost_empires],
179179
)
@@ -190,7 +190,8 @@ def test_filter_unsupported_lookups(self):
190190
def test_len(self):
191191
self.assertCountEqual(MuseumExhibit.objects.filter(sections__len=10), [])
192192
self.assertCountEqual(
193-
MuseumExhibit.objects.filter(sections__len=1), [self.egypt, self.new_descoveries]
193+
MuseumExhibit.objects.filter(sections__len=1),
194+
[self.egypt, self.wonders, self.new_descoveries],
194195
)
195196
# Nested EMF
196197
self.assertCountEqual(
@@ -202,23 +203,23 @@ def test_len(self):
202203
self.assertCountEqual(
203204
MuseumExhibit.objects.filter(sections__0__artifacts__len=2), [self.wonders]
204205
)
205-
self.assertCountEqual(
206-
MuseumExhibit.objects.filter(sections__0__artifacts__len=0), [self.new_descoveries]
207-
)
208-
self.assertCountEqual(
209-
MuseumExhibit.objects.filter(sections__1__artifacts__len=1), [self.wonders]
210-
)
206+
self.assertCountEqual(MuseumExhibit.objects.filter(sections__0__artifacts__len=0), [])
207+
self.assertCountEqual(MuseumExhibit.objects.filter(sections__1__artifacts__len=1), [])
211208

212209
def test_overlap_simplefield(self):
213210
self.assertSequenceEqual(
214211
MuseumExhibit.objects.filter(sections__section_number__overlap=[10]), []
215212
)
216213
self.assertSequenceEqual(
217214
MuseumExhibit.objects.filter(sections__section_number__overlap=[1]),
218-
[self.egypt, self.wonders, self.new_descoveries],
215+
[self.egypt, self.wonders],
216+
)
217+
self.assertSequenceEqual(
218+
MuseumExhibit.objects.filter(sections__section_number__overlap=[2]),
219+
[self.new_descoveries],
219220
)
220221
self.assertSequenceEqual(
221-
MuseumExhibit.objects.filter(sections__section_number__overlap=[2]), [self.wonders]
222+
MuseumExhibit.objects.filter(sections__section_number__overlap=[3]), []
222223
)
223224

224225

0 commit comments

Comments
 (0)