Function _compute_mu

tests/wycheproof/test_mldsa.py:167–172

(pub_raw: bytes, msg: bytes, ctx: bytes)

def _compute_mu(pub_raw: bytes, msg: bytes, ctx: bytes) -> bytes:
    # FIPS 204: mu = SHAKE256(SHAKE256(pk, 64) || M', 64) where for pure
    # ML-DSA M' = 0x00 || len(ctx) || ctx || M.
    tr = hashlib.shake_256(pub_raw).digest(64)
    m_prime = b"\x00" + bytes([len(ctx)]) + ctx + msg
    return hashlib.shake_256(tr + m_prime).digest(64)


_MLDSA_PUBLIC_KEYS: dict[