Skip to content

Commit 1149577

Browse files
author
Ryan P Kilby
committed
Use setUpTestData to improve test speed
1 parent fe72834 commit 1149577

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

rest_framework_filters/tests.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
from dateutil.parser import parse as date_parse
99

10+
import django
1011
from django.db import models
1112
from django.test import TestCase
1213
from django.contrib.auth.models import User
@@ -238,7 +239,18 @@ class Meta:
238239
model = Person
239240

240241
class TestFilterSets(TestCase):
241-
def setUp(self):
242+
243+
if django.VERSION >= (1, 8):
244+
@classmethod
245+
def setUpTestData(cls):
246+
cls.generateTestData()
247+
248+
else:
249+
def setUp(self):
250+
self.generateTestData()
251+
252+
@classmethod
253+
def generateTestData(cls):
242254
#######################
243255
# Create users
244256
#######################

0 commit comments

Comments
 (0)