diff options
| -rw-r--r-- | tests/unittest_inference.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/unittest_inference.py b/tests/unittest_inference.py index 83dfcfb5..91a14ec6 100644 --- a/tests/unittest_inference.py +++ b/tests/unittest_inference.py @@ -1701,7 +1701,9 @@ class InferenceTest(resources.SysPathSetup, unittest.TestCase): pass """ ) - self.assertIn("object", [base.name for base in klass.ancestors()]) + ancestors = [base.name for base in klass.ancestors()] + expected_subset = ["datetime", "date"] + self.assertEqual(expected_subset, ancestors[:2]) def test_stop_iteration_leak(self): code = """ |