liboqs/tests/test_cmdline.py
Spencer Wilson b80240c348
Update example files (#2071)
* Update example files to use ML-KEM and ML-DSA

Signed-off-by: Spencer Wilson <spencer.wilson@uwaterloo.ca>

* Call example_sig_stfl in test_cmdline.py

Signed-off-by: Spencer Wilson <spencer.wilson@uwaterloo.ca>

---------

Signed-off-by: Spencer Wilson <spencer.wilson@uwaterloo.ca>
2025-02-07 15:22:07 -05:00

51 lines
1.6 KiB
Python

# SPDX-License-Identifier: MIT
import helpers
import os
import pytest
import sys
@helpers.filtered_test
@pytest.mark.parametrize('program', ['example_kem', 'example_sig', 'example_sig_stfl'])
def test_examples(program):
helpers.run_subprocess(
[helpers.path_to_executable(program)],
)
@helpers.filtered_test
@pytest.mark.parametrize('kem_name', helpers.available_kems_by_name())
def test_kem(kem_name):
if not(helpers.is_kem_enabled_by_name(kem_name)): pytest.skip('Not enabled')
helpers.run_subprocess(
[helpers.path_to_executable('test_kem'), kem_name],
)
@helpers.filtered_test
@pytest.mark.parametrize('sig_name', helpers.available_sigs_by_name())
def test_sig(sig_name):
if not(helpers.is_sig_enabled_by_name(sig_name)): pytest.skip('Not enabled')
helpers.run_subprocess(
[helpers.path_to_executable('test_sig'), sig_name],
)
@helpers.filtered_test
@pytest.mark.parametrize('sig_stfl_name', helpers.available_sig_stfls_by_name())
def test_sig_stfl(sig_stfl_name):
if not(helpers.is_sig_stfl_enabled_by_name(sig_stfl_name)): pytest.skip('Not enabled')
# Test with KATs apply for XMSS
if sig_stfl_name.startswith("XMSS"):
katfile = helpers.get_katfile("sig_stfl", sig_stfl_name)
if not katfile: pytest.skip("KATs file is missing")
helpers.run_subprocess(
[helpers.path_to_executable('test_sig_stfl'), sig_stfl_name, katfile],
)
else:
helpers.run_subprocess(
[helpers.path_to_executable('test_sig_stfl'), sig_stfl_name],
)
if __name__ == "__main__":
import sys
pytest.main(sys.argv)