from quantcrypt.dss import Dilithium

# First, we create an instance of a DSS algorithm
# and also define a message to be signed.
dss = Dilithium()
message = b'Hello World'

# Next, we generate a PK and SK pair.
public_key, secret_key = dss.keygen()

# Then, we use the secret_key to sign the message.
signature = dss.sign(secret_key, message)

# Finally, we use the public_key to verify the validity of the signature.
dss.verify(public_key, message, signature)