status-im
/
eth2.0-specs
mirror of https://github.com/status-im/eth2.0-specs.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix field iteration crash in ssz typing

This commit is contained in:
protolambda 2019-06-19 02:41:43 +02:00
parent a33c67894e
commit 7cdec746b4
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test_libs/pyspec/eth2spec/utils/ssz/ssz_typing.py
View File

@ -154,11 +154,11 @@ class Container(Series, metaclass=SSZType):
super().__setattr__(name, value) super().__setattr__(name, value)
def __repr__(self): def __repr__(self):
return repr({field: getattr(self, field) for field in self.get_fields()}) return repr({field: getattr(self, field) for field in self.get_fields().keys()})
def __str__(self): def __str__(self):
output = [f'{self.__class__.__name__}'] output = [f'{self.__class__.__name__}']
for field in self.get_fields(): for field in self.get_fields().keys():
output.append(f' {field}: {getattr(self, field)}') output.append(f' {field}: {getattr(self, field)}')
return "\n".join(output) return "\n".join(output)
@ -176,7 +176,7 @@ class Container(Series, metaclass=SSZType):
@classmethod @classmethod
def default(cls): def default(cls):
return cls(**{f: t.default() for f, t in cls.get_fields()}) return cls(**{f: t.default() for f, t in cls.get_fields().items()})
@classmethod @classmethod
def is_fixed_size(cls): def is_fixed_size(cls):

4
test_libs/pyspec/eth2spec/utils/ssz/test_ssz_typing.py
View File

@ -76,6 +76,10 @@ def test_container():
a: uint8 a: uint8
b: uint32 b: uint32
empty = Foo()
assert empty.a == uint8(0)
assert empty.b == uint32(0)
assert issubclass(Foo, Container) assert issubclass(Foo, Container)
assert issubclass(Foo, SSZValue) assert issubclass(Foo, SSZValue)
assert issubclass(Foo, Series) assert issubclass(Foo, Series)