4 files changed, 9 insertions, 9 deletions
diff --git a/tests/migrations/test_graph.py b/tests/migrations/test_graph.py
index 44608c0128..b4252a94c8 100644
--- a/tests/migrations/test_graph.py
+++ b/tests/migrations/test_graph.py
@@ -4,11 +4,11 @@ from django.db.migrations.exceptions import (
     CircularDependencyError, NodeNotFoundError,
 )
 from django.db.migrations.graph import RECURSION_DEPTH_WARNING, MigrationGraph
-from django.test import TestCase
+from django.test import SimpleTestCase
 from django.utils.encoding import force_text
 
 
-class GraphTests(TestCase):
+class GraphTests(SimpleTestCase):
     """
     Tests the digraph structure.
     """
diff --git a/tests/migrations/test_optimizer.py b/tests/migrations/test_optimizer.py
index e639752082..e599baac83 100644
--- a/tests/migrations/test_optimizer.py
+++ b/tests/migrations/test_optimizer.py
@@ -2,12 +2,12 @@
 
 from django.db import migrations, models
 from django.db.migrations.optimizer import MigrationOptimizer
-from django.test import TestCase
+from django.test import SimpleTestCase
 
 from .models import CustomModelBase, EmptyManager
 
 
-class OptimizerTests(TestCase):
+class OptimizerTests(SimpleTestCase):
     """
     Tests the migration autodetector.
     """
diff --git a/tests/migrations/test_state.py b/tests/migrations/test_state.py
index 3c6abe3a91..2ad8025122 100644
--- a/tests/migrations/test_state.py
+++ b/tests/migrations/test_state.py
@@ -7,7 +7,7 @@ from django.db.migrations.operations import (
 from django.db.migrations.state import (
     ModelState, ProjectState, get_related_models_recursive,
 )
-from django.test import SimpleTestCase, TestCase, override_settings
+from django.test import SimpleTestCase, override_settings
 from django.utils import six
 
 from .models import (
@@ -16,7 +16,7 @@ from .models import (
 )
 
 
-class StateTests(TestCase):
+class StateTests(SimpleTestCase):
     """
     Tests state construction, rendering and modification by operations.
     """
@@ -750,7 +750,7 @@ class StateTests(TestCase):
         self.assertEqual(list(choices_field.choices), choices)
 
 
-class ModelStateTests(TestCase):
+class ModelStateTests(SimpleTestCase):
     def test_custom_model_base(self):
         state = ModelState.from_model(ModelWithCustomBase)
         self.assertEqual(state.bases, (models.Model,))
diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py
index f0edfd3e30..ef9489748d 100644
--- a/tests/migrations/test_writer.py
+++ b/tests/migrations/test_writer.py
@@ -17,7 +17,7 @@ from django.db import migrations, models
 from django.db.migrations.writer import (
     MigrationWriter, OperationWriter, SettingsReference,
 )
-from django.test import SimpleTestCase, TestCase, ignore_warnings
+from django.test import SimpleTestCase, ignore_warnings
 from django.utils import datetime_safe, six
 from django.utils._os import upath
 from django.utils.deconstruct import deconstructible
@@ -152,7 +152,7 @@ class OperationWriterTests(SimpleTestCase):
         )
 
 
-class WriterTests(TestCase):
+class WriterTests(SimpleTestCase):
     """
     Tests the migration writer (makes migration files from Migration instances)
     """