diff --git a/cookbook/tests/test_setup.py b/cookbook/tests/test_setup.py index f5447e20..7f9ba59c 100644 --- a/cookbook/tests/test_setup.py +++ b/cookbook/tests/test_setup.py @@ -1,6 +1,6 @@ from django.contrib import auth -from django.contrib.auth.models import User, Group -from django.test import TestCase, Client +from django.contrib.auth.models import Group, User +from django.test import Client, TestCase class TestBase(TestCase): @@ -38,8 +38,14 @@ class TestBase(TestCase): user.is_superuser = True user.save() - def batch_requests(self, clients, url, method='get', payload={}, content_type=''): + def batch_requests( + self, clients, url, method='get', payload={}, content_type='' + ): for c in clients: if method == 'get': r = c[0].get(url) - self.assertEqual(r.status_code, c[1], msg=f'GET request failed for user {auth.get_user(c[0])} when testing url {url}') + self.assertEqual( + r.status_code, + c[1], + msg=f'GET request failed for user {auth.get_user(c[0])} when testing url {url}' # noqa: E501 + )