status-im/eth2.0-specs
2
0
Fork 0
mirror of https://github.com/status-im/eth2.0-specs.git synced 2025-01-11 19:24:15 +00:00
Code Issues Projects Releases Wiki Activity

Make config dict compatible with yaml.dump

Browse Source
This commit is contained in:
Hsiao-Wei Wang 2021-12-02 00:46:19 +08:00
parent d7f6a42729
commit bb8168fd95
No known key found for this signature in database
GPG Key ID: 1111A8A81778319E
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tests/core/pyspec/eth2spec/test/context.py
View File

@ -1,6 +1,7 @@
import pytest import pytest
from copy import deepcopy from copy import deepcopy
from dataclasses import dataclass from dataclasses import dataclass
from eth_utils import encode_hex
from eth2spec.phase0 import mainnet as spec_phase0_mainnet, minimal as spec_phase0_minimal from eth2spec.phase0 import mainnet as spec_phase0_mainnet, minimal as spec_phase0_minimal
from eth2spec.altair import mainnet as spec_altair_mainnet, minimal as spec_altair_minimal from eth2spec.altair import mainnet as spec_altair_mainnet, minimal as spec_altair_minimal
@ -464,6 +465,22 @@ def with_presets(preset_bases, reason=None):
return decorator return decorator
def _get_basic_dict(ssz_dict: Dict[str, Any]) -> Dict[str, Any]:
"""
Get dict of Python built-in types from a dict of SSZ objects.
"""
result = {}
for k, v in ssz_dict.items():
if isinstance(v, int):
value = int(v)
elif isinstance(v, bytes):
value = encode_hex(v)
else:
value = str(v)
result[k] = value
return result
def with_config_overrides(config_overrides): def with_config_overrides(config_overrides):
""" """
WARNING: the spec_test decorator must wrap this, to ensure the decorated test actually runs. WARNING: the spec_test decorator must wrap this, to ensure the decorated test actually runs.
@ -484,9 +501,10 @@ def with_config_overrides(config_overrides):
# Retain types of all config values # Retain types of all config values
test_config = {k: config_types[k](v) for k, v in tmp_config.items()} test_config = {k: config_types[k](v) for k, v in tmp_config.items()}
# FIXME: config YAML encoding issue # To output the changed config to could be serialized with yaml test vectors,
# Output the config for test vectors # the dict SSZ objects have to be converted into Python built-in types.
# yield 'config', 'data', test_config output_config = _get_basic_dict(test_config)
yield 'config', 'data', output_config
spec.config = spec.Configuration(**test_config) spec.config = spec.Configuration(**test_config)