diff options
author | Serhiy Storchaka <storchaka@gmail.com> | 2021-12-05 20:42:50 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-12-05 20:42:50 (GMT) |
commit | 446be166861b2f08f87f74018113dd98ca5fca02 (patch) | |
tree | 72eed67ccf609faefac71afa67589866a3818843 /Lib | |
parent | 1fd4de5bddbbf2a97cdbac4d298c89e1156bdc6c (diff) | |
download | cpython-446be166861b2f08f87f74018113dd98ca5fca02.zip cpython-446be166861b2f08f87f74018113dd98ca5fca02.tar.gz cpython-446be166861b2f08f87f74018113dd98ca5fca02.tar.bz2 |
bpo-45663: Fix is_dataclass() for dataclasses which are subclasses of types.GenericAlias (GH-29294)
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/dataclasses.py | 2 | ||||
-rw-r--r-- | Lib/test/test_dataclasses.py | 12 |
2 files changed, 13 insertions, 1 deletions
diff --git a/Lib/dataclasses.py b/Lib/dataclasses.py index da06aa6..3f85d85 100644 --- a/Lib/dataclasses.py +++ b/Lib/dataclasses.py @@ -1211,7 +1211,7 @@ def _is_dataclass_instance(obj): def is_dataclass(obj): """Returns True if obj is a dataclass or an instance of a dataclass.""" - cls = obj if isinstance(obj, type) else type(obj) + cls = obj if isinstance(obj, type) and not isinstance(obj, GenericAlias) else type(obj) return hasattr(cls, _FIELDS) diff --git a/Lib/test/test_dataclasses.py b/Lib/test/test_dataclasses.py index 47075df..ef5009a 100644 --- a/Lib/test/test_dataclasses.py +++ b/Lib/test/test_dataclasses.py @@ -8,6 +8,7 @@ import abc import pickle import inspect import builtins +import types import unittest from unittest.mock import Mock from typing import ClassVar, Any, List, Union, Tuple, Dict, Generic, TypeVar, Optional, Protocol @@ -1354,6 +1355,17 @@ class TestCase(unittest.TestCase): with self.assertRaisesRegex(TypeError, 'should be called on dataclass instances'): replace(obj, x=0) + def test_is_dataclass_genericalias(self): + @dataclass + class A(types.GenericAlias): + origin: type + args: type + self.assertTrue(is_dataclass(A)) + a = A(list, int) + self.assertTrue(is_dataclass(type(a))) + self.assertTrue(is_dataclass(a)) + + def test_helper_fields_with_class_instance(self): # Check that we can call fields() on either a class or instance, # and get back the same thing. |