|
1 | 1 | from __future__ import unicode_literals
|
2 | 2 |
|
3 | 3 | from datetime import datetime, timedelta
|
4 |
| -try: |
5 |
| - from unittest import skipUnless |
6 |
| -except ImportError: |
7 |
| - from unittest2 import skipUnless |
| 4 | +import unittest |
8 | 5 |
|
9 | 6 | import django
|
10 |
| -try: |
11 |
| - from django.contrib.auth import get_user_model |
12 |
| - User = get_user_model() |
13 |
| -except ImportError: # django 1.4 compatibility |
14 |
| - from django.contrib.auth.models import User |
15 | 7 | from django.db import models
|
16 | 8 | from django.db.models.loading import get_model
|
17 | 9 | from django.db.models.fields.proxy import OrderWrt
|
|
29 | 21 | )
|
30 | 22 | from ..external.models import ExternalModel2, ExternalModel4
|
31 | 23 |
|
| 24 | +try: |
| 25 | + from unittest import skipUnless |
| 26 | +except ImportError: |
| 27 | + from unittest2 import skipUnless |
| 28 | +try: |
| 29 | + from django.contrib.auth import get_user_model |
| 30 | + User = get_user_model() |
| 31 | +except ImportError: # django 1.4 compatibility |
| 32 | + from django.contrib.auth.models import User |
| 33 | + |
32 | 34 | today = datetime(2021, 1, 1, 10, 0)
|
33 | 35 | tomorrow = today + timedelta(days=1)
|
34 | 36 | yesterday = today - timedelta(days=1)
|
@@ -324,7 +326,7 @@ def test_register_custome_records(self):
|
324 | 326 | self.assertEqual(expected, str(voter.history.all()[0])[:len(expected)])
|
325 | 327 |
|
326 | 328 |
|
327 |
| -class CreateHistoryModelTests(TestCase): |
| 329 | +class CreateHistoryModelTests(unittest.TestCase): |
328 | 330 |
|
329 | 331 | def test_create_history_model_with_one_to_one_field_to_integer_field(self):
|
330 | 332 | records = HistoricalRecords()
|
|
0 commit comments