diff options
Diffstat (limited to 'tests/model_fields/test_autofield.py')
| -rw-r--r-- | tests/model_fields/test_autofield.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tests/model_fields/test_autofield.py b/tests/model_fields/test_autofield.py index 646cd2ab0a..0ae981c3d9 100644 --- a/tests/model_fields/test_autofield.py +++ b/tests/model_fields/test_autofield.py @@ -30,6 +30,18 @@ class AutoFieldInheritanceTests(SimpleTestCase): self.assertIsInstance(field(), models.AutoField) def test_issubclass_of_autofield(self): - for field in (models.BigAutoField, models.SmallAutoField): + class MyBigAutoField(models.BigAutoField): + pass + + class MySmallAutoField(models.SmallAutoField): + pass + + tests = [ + MyBigAutoField, + MySmallAutoField, + models.BigAutoField, + models.SmallAutoField, + ] + for field in tests: with self.subTest(field.__name__): self.assertTrue(issubclass(field, models.AutoField)) |
