@@ -42,7 +42,7 @@ def _get_shared_prod_key_manager() -> XmssKeyManager:
4242 Returns:
4343 Shared XmssKeyManager instance with max_slot=10.
4444 """
45- return XmssKeyManager (max_slot = Slot (10 ), scheme = PROD_SIGNATURE_SCHEME )
45+ return XmssKeyManager (max_slot = Slot (10 ))
4646
4747
4848class VerifySignaturesTest (BaseConsensusFixture ):
@@ -145,7 +145,7 @@ def make_fixture(self) -> VerifySignaturesTest:
145145
146146 # Verify signatures
147147 try :
148- signed_block .verify_signatures (self .anchor_state , scheme = PROD_SIGNATURE_SCHEME )
148+ signed_block .verify_signatures (self .anchor_state )
149149 except AssertionError as e :
150150 exception_raised = e
151151 # If we expect an exception, this is fine
@@ -236,13 +236,13 @@ def _build_block_from_spec(
236236 proposer_attestation_signature = key_manager .sign_attestation (proposer_attestation )
237237 else :
238238 # Generate an invalid dummy signature (all zeros)
239- from lean_spec .subspecs .xmss .constants import PROD_CONFIG
239+ from lean_spec .subspecs .xmss .constants import TEST_CONFIG
240240 from lean_spec .subspecs .xmss .containers import Signature
241241 from lean_spec .subspecs .xmss .types import HashDigestList , HashTreeOpening , Randomness
242242
243243 proposer_attestation_signature = Signature (
244244 path = HashTreeOpening (siblings = HashDigestList (data = [])),
245- rho = Randomness (data = [Fp (0 ) for _ in range (PROD_CONFIG .RAND_LEN_FE )]),
245+ rho = Randomness (data = [Fp (0 ) for _ in range (TEST_CONFIG .RAND_LEN_FE )]),
246246 hashes = HashDigestList (data = []),
247247 )
248248
@@ -333,13 +333,13 @@ def _build_signed_attestation_from_spec(
333333 signature = key_manager .sign_attestation (attestation )
334334 else :
335335 # Generate an invalid dummy signature (all zeros)
336- from lean_spec .subspecs .xmss .constants import PROD_CONFIG
336+ from lean_spec .subspecs .xmss .constants import TEST_CONFIG
337337 from lean_spec .subspecs .xmss .containers import Signature
338338 from lean_spec .subspecs .xmss .types import HashDigestList , HashTreeOpening , Randomness
339339
340340 signature = Signature (
341341 path = HashTreeOpening (siblings = HashDigestList (data = [])),
342- rho = Randomness (data = [Fp (0 ) for _ in range (PROD_CONFIG .RAND_LEN_FE )]),
342+ rho = Randomness (data = [Fp (0 ) for _ in range (TEST_CONFIG .RAND_LEN_FE )]),
343343 hashes = HashDigestList (data = []),
344344 )
345345
0 commit comments