gh-89683: add tests for deepcopy on frozen dataclasses (gh-123098)

Co-authored-by: Eric V. Smith <ericvsmith@users.noreply.github.com>
This commit is contained in:
Bénédikt Tran 2024-09-26 23:15:28 +02:00 committed by GitHub
parent 2c10832887
commit 5e7eba09bc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -17,6 +17,7 @@ from unittest.mock import Mock
from typing import ClassVar, Any, List, Union, Tuple, Dict, Generic, TypeVar, Optional, Protocol, DefaultDict from typing import ClassVar, Any, List, Union, Tuple, Dict, Generic, TypeVar, Optional, Protocol, DefaultDict
from typing import get_type_hints from typing import get_type_hints
from collections import deque, OrderedDict, namedtuple, defaultdict from collections import deque, OrderedDict, namedtuple, defaultdict
from copy import deepcopy
from functools import total_ordering, wraps from functools import total_ordering, wraps
import typing # Needed for the string "typing.ClassVar[int]" to work as an annotation. import typing # Needed for the string "typing.ClassVar[int]" to work as an annotation.
@ -3175,6 +3176,48 @@ class TestFrozen(unittest.TestCase):
with self.assertRaisesRegex(TypeError, 'unhashable type'): with self.assertRaisesRegex(TypeError, 'unhashable type'):
hash(C({})) hash(C({}))
def test_frozen_deepcopy_without_slots(self):
# see: https://github.com/python/cpython/issues/89683
@dataclass(frozen=True, slots=False)
class C:
s: str
c = C('hello')
self.assertEqual(deepcopy(c), c)
def test_frozen_deepcopy_with_slots(self):
# see: https://github.com/python/cpython/issues/89683
with self.subTest('generated __slots__'):
@dataclass(frozen=True, slots=True)
class C:
s: str
c = C('hello')
self.assertEqual(deepcopy(c), c)
with self.subTest('user-defined __slots__ and no __{get,set}state__'):
@dataclass(frozen=True, slots=False)
class C:
__slots__ = ('s',)
s: str
# with user-defined slots, __getstate__ and __setstate__ are not
# automatically added, hence the error
err = r"^cannot\ assign\ to\ field\ 's'$"
self.assertRaisesRegex(FrozenInstanceError, err, deepcopy, C(''))
with self.subTest('user-defined __slots__ and __{get,set}state__'):
@dataclass(frozen=True, slots=False)
class C:
__slots__ = ('s',)
__getstate__ = dataclasses._dataclass_getstate
__setstate__ = dataclasses._dataclass_setstate
s: str
c = C('hello')
self.assertEqual(deepcopy(c), c)
class TestSlots(unittest.TestCase): class TestSlots(unittest.TestCase):
def test_simple(self): def test_simple(self):