Refactor unit test

Author: WaVEV

tests/indexes_/test_atlas_indexes.py

@@ -1,5 +1,5 @@
 from django.db import connection
-from django.test import TestCase, skipIfDBFeature, skipUnlessDBFeature
+from django.test import SimpleTestCase, TestCase, skipIfDBFeature, skipUnlessDBFeature
 
 from django_mongodb_backend.indexes import SearchIndex, VectorSearchIndex
 
@@ -26,20 +26,8 @@ def assertAddRemoveIndex(self, editor, model, index):
         )
 
 
-class SearchIndexTests(TestMixin, TestCase):
-    """Tests for creating, validating, and removing search indexes using Django's schema editor."""
-
-    @skipUnlessDBFeature("supports_atlas_search")
-    def test_simple(self):
-        with connection.schema_editor() as editor:
-            index = SearchIndex(
-                name="recent_article_idx",
-                fields=["number"],
-            )
-            editor.add_index(index=index, model=Article)
-            self.assertAddRemoveIndex(editor, Article, index)
-
-    @skipIfDBFeature("supports_atlas_search")
+@skipIfDBFeature("supports_atlas_search")
+class UnsupportedSearchIndexesTests(TestCase):
     def test_index_not_created(self):
         with connection.schema_editor() as editor:
             index = SearchIndex(
@@ -55,7 +43,20 @@ def test_index_not_created(self):
                 ),
             )
 
-    @skipUnlessDBFeature("supports_atlas_search")
+
+@skipUnlessDBFeature("supports_atlas_search")
+class SearchIndexTests(TestMixin, TestCase):
+    """Tests for creating, validating, and removing search indexes using Django's schema editor."""
+
+    def test_simple(self):
+        with connection.schema_editor() as editor:
+            index = SearchIndex(
+                name="recent_article_idx",
+                fields=["number"],
+            )
+            editor.add_index(index=index, model=Article)
+            self.assertAddRemoveIndex(editor, Article, index)
+
     def test_multiple_fields(self):
         with connection.schema_editor() as editor:
             index = SearchIndex(
@@ -98,11 +99,12 @@ def test_multiple_fields(self):
             self.assertAddRemoveIndex(editor, Article, index)
 
 
-class VectorSearchIndexTests(TestMixin, TestCase):
-    """
-    Tests for creating, validating, and removing vector search indexes
-    using Django's schema editor.
-    """
+class VectorSearchIndexesInvalidArgumentsTests(SimpleTestCase):  # TODO CHANGE NAME
+    def test_deconstruct_default_similarity(self):
+        index = VectorSearchIndex(name="recent_article_idx", fields=["number"])
+        name, args, kwargs = index.deconstruct()
+        new = VectorSearchIndex(*args, **kwargs)
+        self.assertEqual(new.similarities, index.similarities)
 
     def test_invalid_similarity_function(self):
         msg = (
@@ -123,14 +125,9 @@ def test_define_field_twice(self):
                 similarities="dotProduct",
             )
 
-    @skipUnlessDBFeature("supports_atlas_search")
-    def test_deconstruct_default_similarity(self):
-        index = VectorSearchIndex(name="recent_article_idx", fields=["number"])
-        name, args, kwargs = index.deconstruct()
-        new = VectorSearchIndex(*args, **kwargs)
-        self.assertEqual(new.similarities, index.similarities)
 
-    @skipIfDBFeature("supports_atlas_search")
+@skipIfDBFeature("supports_atlas_search")
+class UnsupportedVectorSearchIndexesTests(TestCase):
     def test_index_not_created(self):
         with connection.schema_editor() as editor:
             index = VectorSearchIndex(
@@ -146,7 +143,14 @@ def test_index_not_created(self):
                 ),
             )
 
-    @skipUnlessDBFeature("supports_atlas_search")
+
+@skipUnlessDBFeature("supports_atlas_search")
+class VectorSearchIndexTests(TestMixin, TestCase):
+    """
+    Tests for creating, validating, and removing vector search indexes
+    using Django's schema editor.
+    """
+
     def test_deconstruct_with_similarities(self):
         index = VectorSearchIndex(
             name="recent_article_idx",
@@ -157,7 +161,6 @@ def test_deconstruct_with_similarities(self):
         new = VectorSearchIndex(*args, **kwargs)
         self.assertEqual(new.similarities, index.similarities)
 
-    @skipUnlessDBFeature("supports_atlas_search")
     def test_simple_vector_search(self):
         with connection.schema_editor() as editor:
             index = VectorSearchIndex(
@@ -167,7 +170,6 @@ def test_simple_vector_search(self):
             editor.add_index(index=index, model=Article)
             self.assertAddRemoveIndex(editor, Article, index)
 
-    @skipUnlessDBFeature("supports_atlas_search")
     def test_multiple_fields(self):
         with connection.schema_editor() as editor:
             index = VectorSearchIndex(