Emit ssz_static tests for SSZ StableContainer / Profile

ethereum · Jul 26, 2024 · 5fefd7b · 5fefd7b
1 parent 4a15496
commit 5fefd7b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/generators/ssz_static/main.py b/tests/generators/ssz_static/main.py
@@ -7,7 +7,7 @@
 from eth2spec.debug import random_value, encode
 from eth2spec.test.helpers.constants import TESTGEN_FORKS, MINIMAL, MAINNET
 from eth2spec.test.context import spec_targets
-from eth2spec.utils.ssz.ssz_typing import Container
+from eth2spec.utils.ssz.ssz_typing import Container, Profile, StableContainer
 from eth2spec.utils.ssz.ssz_impl import (
  hash_tree_root,
  serialize,
@@ -32,7 +32,12 @@ def create_test_case(rng: Random, typ,
 def get_spec_ssz_types(spec):
  return [
  (name, value) for (name, value) in getmembers(spec, isclass)
- if issubclass(value, Container) and value != Container # only the subclasses, not the imported base class
+ if (
+ # only the subclasses, not the imported base class
+ issubclass(value, Container) and value != Container
+ or issubclass(value, StableContainer) and value != StableContainer
+ or issubclass(value, Profile) and value != Profile
+ )
  ]