jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-09 10:59:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow inheriting FrozenOrThawed with custom init (#105624)

Browse Source
This commit is contained in:
Erik Montnemery
2023-12-13 10:13:34 +01:00
committed by GitHub
parent c318445a76
commit 22c3847c0e
3 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/helpers/test_entity.py
View File

@@ -1749,3 +1749,15 @@ def test_extending_entity_description(snapshot: SnapshotAssertion):
key="blah", extra="foo", mixin="mixin", name="name"
)
assert repr(obj) == snapshot
# Try inheriting with custom init
@dataclasses.dataclass
class CustomInitEntityDescription(entity.EntityDescription):
def __init__(self, extra, *args, **kwargs) -> None:
super().__init__(*args, **kwargs)
self.extra: str = extra
obj = CustomInitEntityDescription(key="blah", extra="foo", name="name")
assert obj == snapshot
assert obj == CustomInitEntityDescription(key="blah", extra="foo", name="name")
assert repr(obj) == snapshot