armor_names method now accepts PQType enum as optional argument

Author: aabmets

quantcrypt/internal/constants.py

@@ -113,8 +113,9 @@ def iterate(cls) -> Iterator[AlgoSpec]:
                 yield value
 
     @classmethod
-    def armor_names(cls) -> list[str]:
-        return [spec.armor_name() for spec in cls.iterate()]
+    def armor_names(cls, pqa_type: PQAType | None = None) -> list[str]:
+        return [spec.armor_name() for spec in cls.iterate()
+            if not pqa_type or spec.type == pqa_type]
 
 
 SupportedVariants = [PQAVariant.REF]

tests/test_constants.py

@@ -108,9 +108,10 @@ def test_supported_algos():
     for item in const.SupportedAlgos.iterate():
         assert isinstance(item, const.AlgoSpec)
 
-    armor_names = const.SupportedAlgos.armor_names()
-    assert isinstance(armor_names, list)
-    assert all([isinstance(n, str) for n in armor_names])
+    for pqa_type in [None, *const.PQAType.members()]:
+        armor_names = const.SupportedAlgos.armor_names(pqa_type)
+        assert isinstance(armor_names, list)
+        assert all([isinstance(n, str) for n in armor_names])
 
 
 def test_pqclean_repo_archive_url():