From 6a975ee66f074e678becd1320870cf4108bbf015 Mon Sep 17 00:00:00 2001 From: Iurii Kemaev <6885137+hbq1@users.noreply.github.com> Date: Tue, 6 Apr 2021 06:14:01 +0100 Subject: [PATCH] bpo-43176: Fix processing of empty dataclasses (GH-24484) When a dataclass inherits from an empty base, all immutability checks are omitted. This PR fixes this and adds tests for it. Automerge-Triggered-By: GH:ericvsmith (cherry picked from commit 376ffc6ac491da74920aed1b8e35bc371cb766ac) Co-authored-by: Iurii Kemaev <6885137+hbq1@users.noreply.github.com> --- Lib/dataclasses.py | 2 +- Lib/test/test_dataclasses.py | 24 +++++++++++++++++++ .../2021-02-09-07-24-29.bpo-43176.bocNQn.rst | 1 + 3 files changed, 26 insertions(+), 1 deletion(-) create mode 100644 Misc/NEWS.d/next/Library/2021-02-09-07-24-29.bpo-43176.bocNQn.rst diff --git a/Lib/dataclasses.py b/Lib/dataclasses.py index 21c7e0c9e2871a..e015254df78b47 100644 --- a/Lib/dataclasses.py +++ b/Lib/dataclasses.py @@ -836,7 +836,7 @@ def _process_class(cls, init, repr, eq, order, unsafe_hash, frozen): # Only process classes that have been processed by our # decorator. That is, they have a _FIELDS attribute. base_fields = getattr(b, _FIELDS, None) - if base_fields: + if base_fields is not None: has_dataclass_bases = True for f in base_fields.values(): fields[f.name] = f diff --git a/Lib/test/test_dataclasses.py b/Lib/test/test_dataclasses.py index 31caae9d2f3936..e65db176d9d34f 100644 --- a/Lib/test/test_dataclasses.py +++ b/Lib/test/test_dataclasses.py @@ -2568,6 +2568,30 @@ class D(C): self.assertEqual(d.i, 0) self.assertEqual(d.j, 10) + def test_inherit_nonfrozen_from_empty_frozen(self): + @dataclass(frozen=True) + class C: + pass + + with self.assertRaisesRegex(TypeError, + 'cannot inherit non-frozen dataclass from a frozen one'): + @dataclass + class D(C): + j: int + + def test_inherit_nonfrozen_from_empty(self): + @dataclass + class C: + pass + + @dataclass + class D(C): + j: int + + d = D(3) + self.assertEqual(d.j, 3) + self.assertIsInstance(d, C) + # Test both ways: with an intermediate normal (non-dataclass) # class and without an intermediate class. def test_inherit_nonfrozen_from_frozen(self): diff --git a/Misc/NEWS.d/next/Library/2021-02-09-07-24-29.bpo-43176.bocNQn.rst b/Misc/NEWS.d/next/Library/2021-02-09-07-24-29.bpo-43176.bocNQn.rst new file mode 100644 index 00000000000000..66a175d21255c2 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2021-02-09-07-24-29.bpo-43176.bocNQn.rst @@ -0,0 +1 @@ +Fixed processing of empty dataclasses. \ No newline at end of file