values method of SupportedAlgos class now supports filtering by PQAType

Author: aabmets

quantcrypt/internal/constants.py

@@ -116,8 +116,11 @@ def items(cls) -> list[tuple[str, AlgoSpec]]:
         return [(k, v) for k, v in vars(cls).items() if isinstance(v, AlgoSpec)]
 
     @classmethod
-    def values(cls) -> list[AlgoSpec]:
-        return [v for v in vars(cls).values() if isinstance(v, AlgoSpec)]
+    def values(cls, pqa_type: PQAType | None = None) -> list[AlgoSpec]:
+        return [
+            v for v in vars(cls).values() if isinstance(v, AlgoSpec)
+            and (not pqa_type or pqa_type == v.type)
+        ]
 
 
 KDFContext = b"quantcrypt"

tests/test_constants.py

@@ -112,6 +112,9 @@ def test_supported_algos():
         assert isinstance(item[1], const.AlgoSpec)
     for item in const.SupportedAlgos.values():
         assert isinstance(item, const.AlgoSpec)
+    for pqa_type in const.PQAType.members():
+        for item in const.SupportedAlgos.values(pqa_type):
+            assert item.type == pqa_type
 
 
 def test_pqclean_repo_archive_url():