add assertions about v4 test vectors

Author: purificant

tests/conftest.py

@@ -6,6 +6,15 @@
 import pytest
 
 
+@pytest.fixture
+def get_all_test_vectors() -> dict:
+    """Return all used test vectors."""
+    return {
+        "v2": get_test_vector("v2"),
+        "v4": get_test_vector("v4"),
+    }
+
+
 @pytest.fixture
 def get_test_vectors_v2() -> dict:
     """Return spec for v2 test vectors."""

tests/protocol/test_test_vectors.py

@@ -4,17 +4,25 @@
 
 
 @pytest.mark.parametrize(
-    "test_name_startswith,expected_count", [("", 12), ("2-E-", 9), ("2-S-", 3)]
+    "version,test_name_startswith,expected_count",
+    [
+        ("v2", "", 12),
+        ("v2", "2-E-", 9),
+        ("v2", "2-S-", 3),
+        ("v4", "", 12),
+        ("v4", "4-E-", 9),
+        ("v4", "4-S-", 3),
+    ],
 )
 def test_expected_number_of_test_vectors(
-    get_test_vectors_v2, test_name_startswith: str, expected_count: int
+    get_all_test_vectors, version: str, test_name_startswith: str, expected_count: int
 ):
     """Test that official test vector specification matches expectations."""
 
     assert (
         sum(
             1
-            for test_case in get_test_vectors_v2["tests"]
+            for test_case in get_all_test_vectors[version]["tests"]
             if test_case["name"].startswith(test_name_startswith)
         )
         == expected_count