bitcoinpqc 0.4.1

Tapscript signature algorithms for Bitcoin P2MR (BIP 360): secp256k1 Schnorr, ML-DSA-44, and SLH-DSA-SHA2-128s
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

"""BIP-360 P2MR construction e2e against living vectors under tests/vectors/p2mr/.

`p2mr_construction.json` is the shared multi-language bar (pure construction ref).
`p2mr_pqc_construction.json` is preserved from BIPs PR #2202 for rust-bitcoin / workshop
stacks; it is not required to pass the pure-Python ref in CI.
"""

from __future__ import annotations

import json
import sys
import unittest
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent))

from p2mr.p2mr import (  # noqa: E402
    BIP360_tests,
    default_fixtures_dir,
    run_single_test,
)


class TestP2mrConstruction(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        cls.fixtures = Path(default_fixtures_dir())
        cls.construction = cls.fixtures / "p2mr_construction.json"
        cls.pqc = cls.fixtures / "p2mr_pqc_construction.json"
        assert cls.construction.is_file(), f"missing {cls.construction}"
        assert cls.pqc.is_file(), f"missing {cls.pqc}"

    def _run_fixture(self, path: Path) -> None:
        with path.open(encoding="utf-8") as f:
            data = json.load(f)
        self.assertEqual(data.get("version"), 1)
        vectors = data["test_vectors"]
        self.assertGreater(len(vectors), 0)
        failed = []
        for i, v in enumerate(vectors):
            if not run_single_test(v, i + 1):
                failed.append(v.get("id", str(i)))
        self.assertEqual(failed, [], f"failed vectors in {path.name}: {failed}")

    def test_p2mr_construction_json(self) -> None:
        """Living multi-language construction vectors (must pass)."""
        self._run_fixture(self.construction)

    def test_p2mr_pqc_fixture_present(self) -> None:
        """PQC construction JSON is preserved as living data (workshop / rust-bitcoin)."""
        with self.pqc.open(encoding="utf-8") as f:
            data = json.load(f)
        self.assertEqual(data.get("version"), 1)
        self.assertGreater(len(data["test_vectors"]), 0)

    def test_bip360_tests_helper(self) -> None:
        self.assertEqual(BIP360_tests(str(self.construction)), 0)


if __name__ == "__main__":
    unittest.main()