Skip to content

Commit

Permalink
Fix unittest assert calls for Python 3.12
Browse files Browse the repository at this point in the history
  • Loading branch information
cclauss committed Jan 1, 2024
1 parent 6f20d61 commit 3a566a5
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
4 changes: 2 additions & 2 deletions test_haystack/test_managers.py
Original file line number Diff line number Diff line change
Expand Up @@ -242,11 +242,11 @@ def spelling_suggestion(self):

def test_values(self):
sqs = self.search_index.objects.auto_query("test").values("id")
self.assert_(isinstance(sqs, ValuesSearchQuerySet))
self.assertIsInstance(sqs, ValuesSearchQuerySet)

def test_valueslist(self):
sqs = self.search_index.objects.auto_query("test").values_list("id")
self.assert_(isinstance(sqs, ValuesListSearchQuerySet))
self.assertIsInstance(sqs, ValuesListSearchQuerySet)


class CustomManagerTestCase(TestCase):
Expand Down
20 changes: 10 additions & 10 deletions test_haystack/test_query.py
Original file line number Diff line number Diff line change
Expand Up @@ -442,7 +442,7 @@ def test_len(self):
def test_repr(self):
reset_search_queries()
self.assertEqual(len(connections["default"].queries), 0)
self.assertRegexpMatches(
self.assertRegex(
repr(self.msqs),
r"^<SearchQuerySet: query=<test_haystack.mocks.MockSearchQuery object"
r" at 0x[0-9A-Fa-f]+>, using=None>$",
Expand Down Expand Up @@ -967,18 +967,18 @@ def test_or_and(self):
class ValuesQuerySetTestCase(SearchQuerySetTestCase):
def test_values_sqs(self):
sqs = self.msqs.auto_query("test").values("id")
self.assert_(isinstance(sqs, ValuesSearchQuerySet))
self.assertIsInstance(sqs, ValuesSearchQuerySet)

# We'll do a basic test to confirm that slicing works as expected:
self.assert_(isinstance(sqs[0], dict))
self.assert_(isinstance(sqs[0:5][0], dict))
self.assertIsInstance(sqs[0], dict)
self.assertIsInstance(sqs[0:5][0], dict)

def test_valueslist_sqs(self):
sqs = self.msqs.auto_query("test").values_list("id")

self.assert_(isinstance(sqs, ValuesListSearchQuerySet))
self.assert_(isinstance(sqs[0], (list, tuple)))
self.assert_(isinstance(sqs[0:1][0], (list, tuple)))
self.assertIsInstance(sqs, ValuesListSearchQuerySet)
self.assertIsInstance(sqs[0], (list, tuple))
self.assertIsInstance(sqs[0:1][0], (list, tuple))

self.assertRaises(
TypeError,
Expand All @@ -989,12 +989,12 @@ def test_valueslist_sqs(self):
)

flat_sqs = self.msqs.auto_query("test").values_list("id", flat=True)
self.assert_(isinstance(sqs, ValuesListSearchQuerySet))
self.assertIsInstance(sqs, ValuesListSearchQuerySet)

# Note that this will actually be None because a mocked sqs lacks
# anything else:
self.assert_(flat_sqs[0] is None)
self.assert_(flat_sqs[0:1][0] is None)
self.assertIsNone(flat_sqs[0])
self.assertIsNone(flat_sqs[0:1][0])


class EmptySearchQuerySetTestCase(TestCase):
Expand Down

0 comments on commit 3a566a5

Please sign in to comment.