diff --git a/tests/tests.py b/tests/tests.py index f99f8e26..5ad67a55 100644 --- a/tests/tests.py +++ b/tests/tests.py @@ -2,6 +2,7 @@ from functools import reduce import operator import re +from pickle import PicklingError import platform import unittest from unittest import mock @@ -659,7 +660,7 @@ def test_366(self): def test_385(self): Client.objects.create(name='Client Name') - with self.assertRaisesRegex(AttributeError, "local object"): + with self.assertRaisesRegex((AttributeError, PicklingError), "local object"): Client.objects.filter(name='Client Name').cache().first() invalidate_model(Client)