Skip to content

Commit 11d88a0

Browse files
committed
fix: remove remaining prod references
1 parent 448ef33 commit 11d88a0

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

packages/testing/src/consensus_testing/test_fixtures/verify_signatures.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

4848
class 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

Comments
 (0)