Skip to content

Commit 8e83ad1

Browse files
committed
Merge pull request #34 from rpkilby/test-data-setup
Use setUpTestData to improve test speed
2 parents bc16118 + 1149577 commit 8e83ad1

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 django.utils.dateparse import parse_time, parse_datetime
99

10+
import django
1011
from django.db import models
1112
from django.test import TestCase
1213
from django.contrib.auth.models import User
@@ -244,7 +245,18 @@ class Meta:
244245
model = Person
245246

246247
class TestFilterSets(TestCase):
247-
def setUp(self):
248+
249+
if django.VERSION >= (1, 8):
250+
@classmethod
251+
def setUpTestData(cls):
252+
cls.generateTestData()
253+
254+
else:
255+
def setUp(self):
256+
self.generateTestData()
257+
258+
@classmethod
259+
def generateTestData(cls):
248260
#######################
249261
# Create users
250262
#######################

0 commit comments

Comments
 (0)