summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorSerhiy Storchaka <storchaka@gmail.com>2021-12-05 20:42:50 (GMT)
committerGitHub <noreply@github.com>2021-12-05 20:42:50 (GMT)
commit446be166861b2f08f87f74018113dd98ca5fca02 (patch)
tree72eed67ccf609faefac71afa67589866a3818843 /Lib
parent1fd4de5bddbbf2a97cdbac4d298c89e1156bdc6c (diff)
downloadcpython-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.py2
-rw-r--r--Lib/test/test_dataclasses.py12
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.