diff options
| -rw-r--r-- | tests/serializers/tests.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/serializers/tests.py b/tests/serializers/tests.py index 987a887f4b..ba5bb34434 100644 --- a/tests/serializers/tests.py +++ b/tests/serializers/tests.py @@ -487,8 +487,8 @@ class NoYamlSerializerTestCase(TestCase): def test_dumpdata_pyyaml_error_message(self): """Calling dumpdata produces an error when yaml package missing""" - self.assertRaisesRegexp(management.CommandError, YAML_IMPORT_ERROR_MESSAGE, - management.call_command, 'dumpdata', format='yaml') + with six.assertRaisesRegex(self, management.CommandError, YAML_IMPORT_ERROR_MESSAGE): + management.call_command('dumpdata', format='yaml') @unittest.skipUnless(HAS_YAML, "No yaml library detected") |