mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-07-28 07:34:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fixed #32492 -- Added TrigramWordSimilarity() and TrigramWordDistance() on PostgreSQL.

Browse source
This commit is contained in:
Nikita Marchant 2021-09-15 12:57:49 +02:00 • committed by Mariusz Felisiak
parent 4ca508a689
commit 4e4082f939
10 changed files with 148 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

39
tests/postgres_tests/test_trigram.py
View file

@ -5,7 +5,8 @@ from .models import CharFieldModel, TextFieldModel
try:
from django.contrib.postgres.search import (
TrigramDistance, TrigramSimilarity,
TrigramDistance, TrigramSimilarity, TrigramWordDistance,
TrigramWordSimilarity,
)
except ImportError:
pass
@ -30,6 +31,15 @@ class TrigramTest(PostgreSQLTestCase):
transform=lambda instance: instance.field,
)
def test_trigram_word_search(self):
obj = self.Model.objects.create(
field='Gumby rides on the path of Middlesbrough',
)
self.assertSequenceEqual(
self.Model.objects.filter(field__trigram_word_similar='Middlesborough'),
[obj],
)
def test_trigram_similarity(self):
search = 'Bat sat on cat.'
# Round result of similarity because PostgreSQL 12+ uses greater
@ -43,6 +53,20 @@ class TrigramTest(PostgreSQLTestCase):
ordered=True,
)
def test_trigram_word_similarity(self):
search = 'mat'
self.assertSequenceEqual(
self.Model.objects.filter(
field__trigram_word_similar=search,
).annotate(
word_similarity=TrigramWordSimilarity(search, 'field'),
).values('field', 'word_similarity').order_by('-word_similarity'),
[
{'field': 'Cat sat on mat.', 'word_similarity': 1.0},
{'field': 'Matthew', 'word_similarity': 0.75},
],
)
def test_trigram_similarity_alternate(self):
# Round result of distance because PostgreSQL 12+ uses greater
# precision.
@ -55,6 +79,19 @@ class TrigramTest(PostgreSQLTestCase):
ordered=True,
)
def test_trigram_word_similarity_alternate(self):
self.assertSequenceEqual(
self.Model.objects.annotate(
word_distance=TrigramWordDistance('mat', 'field'),
).filter(
word_distance__lte=0.7,
).values('field', 'word_distance').order_by('word_distance'),
[
{'field': 'Cat sat on mat.', 'word_distance': 0},
{'field': 'Matthew', 'word_distance': 0.25},
],
)
class TrigramTextFieldTest(TrigramTest):
"""