fixed mypy issues

This commit is contained in:
Tim Riddermann 2023-05-09 11:27:19 +02:00
parent 50a2f34fc6
commit 3a676478ee
2 changed files with 5 additions and 3 deletions

View file

@ -27,6 +27,8 @@ class DataclassMixin:
methods exists and every name can be used.
"""
__dataclass_fields__: t.ClassVar[dict[str, dataclasses.Field[t.Any]]]
def __getitem__(self, key: str) -> t.Any:
return getattr(self, key)
@ -269,4 +271,4 @@ TBitStruct = DataclassBitStruct
TContainerMixin = DataclassMixin
TContainerBase = DataclassMixin
TStructField = csfield
sfield = csfield
sfield = csfield

View file

@ -384,7 +384,7 @@ def test_tenum_no_enumbase() -> None:
assert raises(lambda: cst.TEnum(cs.Byte, cls)) == TypeError
def test_tenum_asdict():
def test_tenum_asdict() -> None:
# see: https://github.com/timrid/construct-typing/issues/21
import construct_typed as cst
import dataclasses
@ -498,7 +498,7 @@ def test_tenum_flags() -> None:
assert raises(d.build, 2) == TypeError
def test_tenum_flags_asdict():
def test_tenum_flags_asdict() -> None:
import construct_typed as cst
import dataclasses