Skip to content

Commit 9e33cff

Browse files
committed
Refactor unit test
1 parent aa6a633 commit 9e33cff

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

tests/indexes_/test_checks.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,10 @@ class Meta:
3939
)
4040

4141

42+
@skipIfDBFeature("supports_atlas_search")
4243
@isolate_apps("indexes_", attr_name="apps")
4344
@override_system_checks([check_vector_search_indexes])
44-
class InvalidVectorSearchIndexesTests(TestCase):
45-
@skipIfDBFeature("supports_atlas_search")
45+
class UnsupportedSearchIndexesTests(TestCase):
4646
def test_requires_atlas_search_support(self):
4747
class Article(models.Model):
4848
title = models.CharField(max_length=10)
@@ -66,7 +66,11 @@ class Meta:
6666
],
6767
)
6868

69-
@skipUnlessDBFeature("supports_atlas_search")
69+
70+
@skipUnlessDBFeature("supports_atlas_search")
71+
@isolate_apps("indexes_", attr_name="apps")
72+
@override_system_checks([check_vector_search_indexes])
73+
class InvalidVectorSearchIndexesTests(TestCase):
7074
def test_requires_size(self):
7175
class Article(models.Model):
7276
title_embedded = ArrayField(models.FloatField())
@@ -86,7 +90,6 @@ class Meta:
8690
],
8791
)
8892

89-
@skipUnlessDBFeature("supports_atlas_search")
9093
def test_requires_float_inner_field(self):
9194
class Article(models.Model):
9295
title_embedded = ArrayField(models.CharField(), size=30)
@@ -106,7 +109,6 @@ class Meta:
106109
],
107110
)
108111

109-
@skipUnlessDBFeature("supports_atlas_search")
110112
def test_unsupported_type(self):
111113
class Article(models.Model):
112114
data = models.JSONField()
@@ -128,7 +130,6 @@ class Meta:
128130
],
129131
)
130132

131-
@skipUnlessDBFeature("supports_atlas_search")
132133
def test_invalid_similarity_function(self):
133134
class Article(models.Model):
134135
vector_data = ArrayField(models.DecimalField(), size=10)
@@ -151,7 +152,6 @@ class Meta:
151152
],
152153
)
153154

154-
@skipUnlessDBFeature("supports_atlas_search")
155155
def test_invalid_similarities_function(self):
156156
class Article(models.Model):
157157
vector1 = ArrayField(models.DecimalField(), size=10)
@@ -185,7 +185,6 @@ class Meta:
185185
],
186186
)
187187

188-
@skipUnlessDBFeature("supports_atlas_search")
189188
def test_define_field_twice(self):
190189
class Article(models.Model):
191190
vector_data = ArrayField(models.DecimalField(), size=10)
@@ -213,7 +212,6 @@ class Meta:
213212
],
214213
)
215214

216-
@skipUnlessDBFeature("supports_atlas_search")
217215
def test_simple(self):
218216
class Article(models.Model):
219217
vector_data = ArrayField(models.DecimalField(), size=10)

0 commit comments

Comments
 (0)