diff --git a/polymorphic/tests.py b/polymorphic/tests.py index ee165ed..379e955 100644 --- a/polymorphic/tests.py +++ b/polymorphic/tests.py @@ -682,11 +682,20 @@ class PolymorphicTests(TestCase): self.assertEqual(show_base_manager(PlainC), " ") self.assertEqual(show_base_manager(Model2A), " ") - self.assertEqual(show_base_manager(Model2B), " ") - self.assertEqual(show_base_manager(Model2C), " ") + if django.VERSION >= (1, 10): + # The new inheritance makes all model levels polymorphic + self.assertEqual(show_base_manager(Model2B), " ") + self.assertEqual(show_base_manager(Model2C), " ") + else: + self.assertEqual(show_base_manager(Model2B), " ") + self.assertEqual(show_base_manager(Model2C), " ") self.assertEqual(show_base_manager(One2OneRelatingModel), " ") - self.assertEqual(show_base_manager(One2OneRelatingModelDerived), " ") + if django.VERSION >= (1, 10): + # The new inheritance makes all model levels polymorphic + self.assertEqual(show_base_manager(One2OneRelatingModelDerived), " ") + else: + self.assertEqual(show_base_manager(One2OneRelatingModelDerived), " ") def test_instance_default_manager(self): def show_default_manager(instance):