Skip to content

Commit 3cfd188

Browse files
WaVEVWaVEV
committed
Remove test with data
1 parent dfa1fb3 commit 3cfd188

File tree

1 file changed

+1
-202
lines changed

1 file changed

+1
-202
lines changed

‎tests/indexes_/test_atlas_indexes.py‎

Lines changed: 1 addition & 202 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,10 @@
1-
import datetime
2-
31
from django.core.exceptions import FieldDoesNotExist
42
from django.db import connection
53
from django.test import TestCase
64

75
from django_mongodb_backend.indexes import SearchIndex, VectorSearchIndex
86

9-
from .models import Article, Data
7+
from .models import Article
108

119

1210
class AtlasIndexTests(TestCase):
@@ -95,97 +93,6 @@ def test_field_not_exists(self):
9593
editor.add_index(index=index, model=Article)
9694

9795

98-
class AtlasIndexTestsWithData(TestCase):
99-
available_apps = None
100-
101-
@classmethod
102-
def setUpTestData(cls):
103-
articles = [
104-
Article(
105-
headline=f"Title {i}",
106-
number=i,
107-
body=f"body {i}",
108-
data="{json: i}",
109-
embedded=Data(integer=i),
110-
auto_now=datetime.datetime.now(),
111-
title_embedded=[0.1] * 10,
112-
description_embedded=[2.5] * 10,
113-
number_list=[2] * i,
114-
name_list=[f"name_{i}"] * 10,
115-
)
116-
for i in range(5)
117-
]
118-
cls.objs = Article.objects.bulk_create(articles)
119-
120-
def assertAddRemoveIndex(self, editor, model, index):
121-
editor.add_index(index=index, model=model)
122-
self.assertIn(
123-
index.name,
124-
connection.introspection.get_constraints(
125-
cursor=None,
126-
table_name=model._meta.db_table,
127-
),
128-
)
129-
editor.remove_index(index=index, model=model)
130-
self.assertNotIn(
131-
index.name,
132-
connection.introspection.get_constraints(
133-
cursor=None,
134-
table_name=model._meta.db_table,
135-
),
136-
)
137-
138-
def test_simple(self):
139-
with connection.schema_editor() as editor:
140-
index = SearchIndex(
141-
name="recent_article_idx",
142-
fields=["number"],
143-
)
144-
editor.add_index(index=index, model=Article)
145-
self.assertAddRemoveIndex(editor, Article, index)
146-
147-
def test_multiple_fields(self):
148-
with connection.schema_editor() as editor:
149-
index = SearchIndex(
150-
name="recent_article_idx",
151-
fields=["headline", "number", "body", "data", "embedded", "auto_now"],
152-
)
153-
editor.add_index(index=index, model=Article)
154-
index_info = connection.introspection.get_constraints(
155-
cursor=None,
156-
table_name=Article._meta.db_table,
157-
)
158-
expected_options = {
159-
"dynamic": False,
160-
"fields": {
161-
"auto_now": {"type": "date"},
162-
"body": {
163-
"indexOptions": "offsets",
164-
"norms": "include",
165-
"store": True,
166-
"type": "string",
167-
},
168-
"data": {"dynamic": False, "fields": {}, "type": "document"},
169-
"embedded": {"dynamic": False, "fields": {}, "type": "embeddedDocuments"},
170-
"headline": {
171-
"indexOptions": "offsets",
172-
"norms": "include",
173-
"store": True,
174-
"type": "string",
175-
},
176-
"number": {
177-
"indexDoubles": True,
178-
"indexIntegers": True,
179-
"representation": "double",
180-
"type": "number",
181-
},
182-
},
183-
}
184-
self.assertCountEqual(index_info[index.name]["columns"], index.fields)
185-
self.assertEqual(index_info[index.name]["options"], expected_options)
186-
self.assertAddRemoveIndex(editor, Article, index)
187-
188-
18996
class SearchIndexTests(TestCase):
19097
# Schema editor is used to create the index to test that it works.
19198
# available_apps = ["indexes"]
@@ -265,111 +172,3 @@ def test_field_not_exists(self):
265172
FieldDoesNotExist, msg
266173
), connection.schema_editor() as editor:
267174
editor.add_index(index=index, model=Article)
268-
269-
270-
class SearchIndexTestsWithData(TestCase):
271-
available_apps = None # could be removed?
272-
273-
@classmethod
274-
def setUpTestData(cls):
275-
articles = [
276-
Article(
277-
headline=f"Title {i}",
278-
number=i,
279-
body=f"body {i}",
280-
data="{json: i}",
281-
embedded=Data(integer=i),
282-
auto_now=datetime.datetime.now(),
283-
title_embedded=[0.1] * 10,
284-
description_embedded=[2.5] * 10,
285-
number_list=[2] * i,
286-
name_list=[f"name_{i}"] * 10,
287-
)
288-
for i in range(5)
289-
]
290-
cls.objs = Article.objects.bulk_create(articles)
291-
292-
def assertAddRemoveIndex(self, editor, model, index):
293-
editor.add_index(index=index, model=model)
294-
self.assertIn(
295-
index.name,
296-
connection.introspection.get_constraints(
297-
cursor=None,
298-
table_name=model._meta.db_table,
299-
),
300-
)
301-
editor.remove_index(index=index, model=model)
302-
self.assertNotIn(
303-
index.name,
304-
connection.introspection.get_constraints(
305-
cursor=None,
306-
table_name=model._meta.db_table,
307-
),
308-
)
309-
310-
def test_simple_vector_search(self):
311-
with connection.schema_editor() as editor:
312-
index = VectorSearchIndex(
313-
name="recent_article_idx",
314-
fields=["number"],
315-
)
316-
editor.add_index(index=index, model=Article)
317-
self.assertAddRemoveIndex(editor, Article, index)
318-
319-
def test_multiple_fields(self):
320-
with connection.schema_editor() as editor:
321-
index = VectorSearchIndex(
322-
name="recent_article_idx",
323-
fields=["headline", "number", "body", "description_embedded"],
324-
)
325-
editor.add_index(index=index, model=Article)
326-
index_info = connection.introspection.get_constraints(
327-
cursor=None,
328-
table_name=Article._meta.db_table,
329-
)
330-
expected_options = {
331-
"latestDefinition": {
332-
"fields": [
333-
{"path": "headline", "type": "filter"},
334-
{"path": "number", "type": "filter"},
335-
{"path": "body", "type": "filter"},
336-
{
337-
"numDimensions": 10,
338-
"path": "description_embedded",
339-
"similarity": "cosine",
340-
"type": "vector",
341-
},
342-
]
343-
},
344-
"latestVersion": 0,
345-
"name": "recent_article_idx",
346-
"queryable": False,
347-
"type": "vectorSearch",
348-
}
349-
self.assertCountEqual(index_info[index.name]["columns"], index.fields)
350-
index_info[index.name]["options"].pop("id")
351-
index_info[index.name]["options"].pop("status")
352-
self.assertEqual(index_info[index.name]["options"], expected_options)
353-
self.assertAddRemoveIndex(editor, Article, index)
354-
355-
def test_corrupted_data(self):
356-
# Mongodb does not raises anything when some vector does not fulfill the required size
357-
with connection.schema_editor() as editor:
358-
Article.objects.create(
359-
headline="Title",
360-
number=10,
361-
body=f"body {10}",
362-
data="{json: 3}",
363-
embedded=Data(integer=3),
364-
auto_now=datetime.datetime.now(),
365-
title_embedded=[0.1] * 15,
366-
description_embedded=[2.5] * 16, # it does not have the required size of ten.
367-
number_list=[2] * 22,
368-
name_list=[f"name_{2}"] * 7,
369-
)
370-
index = VectorSearchIndex(
371-
name="recent_article_idx",
372-
fields=["headline", "number", "body", "description_embedded"],
373-
)
374-
editor.add_index(index=index, model=Article)
375-
self.assertAddRemoveIndex(editor, Article, index)

0 commit comments

Comments
 (0)