mirror of
https://github.com/saymrwulf/autoresearch-quantum.git
synced 2026-05-14 20:37:51 +00:00
saymrwulf
e13a3268c2
- 8 Jupyter notebooks across 3 learning plans (A: bottom-up, B: spiral, C: parallel tracks)
- Teaching toolkit (src/autoresearch_quantum/teaching/) with ipywidgets-based
quiz, predict_choice, reflect, and order widgets — visually distinct from code cells
- Fix spectator_z operator: was {1:'Z',2:'Z'} (IZZI, expectation=0), now {1:'Z',3:'Z'}
(ZIZI, expectation=+1 for ideal T-state, commutes with logical operators)
- Fix u_magic seed: swap phase arguments to match h_p and ry_rz preparations
- Fix double-display bug: widgets rendered twice when function returned the box
- Fix CLI override parser for negative integers and missing '=' validation
- Fix stabilizer detection quiz: ZZZZ detects X errors, not Z errors
- Add ties parameter to order() for questions with interchangeable items
- Expand test suite from 21 to 107 tests
- Update README with notebook instructions and project tree
107 lines
3.8 KiB
Python
107 lines
3.8 KiB
Python
"""Tests for codes.four_two_two — all seed and encoder styles."""
|
|
from __future__ import annotations
|
|
|
|
from math import pi
|
|
|
|
import pytest
|
|
from qiskit.quantum_info import Statevector
|
|
|
|
from autoresearch_quantum.codes.four_two_two import (
|
|
STABILIZERS,
|
|
apply_magic_seed,
|
|
build_encoder,
|
|
build_preparation_circuit,
|
|
encoded_magic_statevector,
|
|
)
|
|
from qiskit import QuantumCircuit
|
|
|
|
|
|
# ── apply_magic_seed ─────────────────────────────────────────────────────────
|
|
|
|
def test_magic_seed_h_p() -> None:
|
|
qc = QuantumCircuit(1)
|
|
apply_magic_seed(qc, 0, "h_p")
|
|
sv = Statevector.from_instruction(qc)
|
|
assert abs(sv.probabilities().sum() - 1.0) < 1e-9
|
|
|
|
|
|
def test_magic_seed_ry_rz() -> None:
|
|
qc = QuantumCircuit(1)
|
|
apply_magic_seed(qc, 0, "ry_rz")
|
|
sv = Statevector.from_instruction(qc)
|
|
assert abs(sv.probabilities().sum() - 1.0) < 1e-9
|
|
|
|
|
|
def test_magic_seed_u_magic() -> None:
|
|
qc = QuantumCircuit(1)
|
|
apply_magic_seed(qc, 0, "u_magic")
|
|
sv = Statevector.from_instruction(qc)
|
|
assert abs(sv.probabilities().sum() - 1.0) < 1e-9
|
|
|
|
|
|
def test_magic_seed_all_styles_equivalent() -> None:
|
|
"""All seed styles should produce the same single-qubit state (up to global phase)."""
|
|
states = []
|
|
for style in ["h_p", "ry_rz", "u_magic"]:
|
|
qc = QuantumCircuit(1)
|
|
apply_magic_seed(qc, 0, style)
|
|
states.append(Statevector.from_instruction(qc))
|
|
# Check fidelity between pairs
|
|
from qiskit.quantum_info import state_fidelity
|
|
for i in range(len(states)):
|
|
for j in range(i + 1, len(states)):
|
|
fid = state_fidelity(states[i], states[j])
|
|
assert fid == pytest.approx(1.0, abs=1e-6), (
|
|
f"Seed styles produced different states: fidelity={fid}"
|
|
)
|
|
|
|
|
|
def test_magic_seed_unsupported_raises() -> None:
|
|
qc = QuantumCircuit(1)
|
|
with pytest.raises(ValueError, match="Unsupported seed style"):
|
|
apply_magic_seed(qc, 0, "unknown_style")
|
|
|
|
|
|
# ── build_encoder ────────────────────────────────────────────────────────────
|
|
|
|
def test_encoder_cx_chain() -> None:
|
|
enc = build_encoder("cx_chain")
|
|
assert enc.num_qubits == 4
|
|
|
|
|
|
def test_encoder_cz_compiled() -> None:
|
|
enc = build_encoder("cz_compiled")
|
|
assert enc.num_qubits == 4
|
|
|
|
|
|
def test_encoder_styles_equivalent() -> None:
|
|
"""Both encoder styles should produce equivalent encoded states."""
|
|
from qiskit.quantum_info import state_fidelity
|
|
sv_cx = Statevector.from_instruction(build_preparation_circuit("h_p", "cx_chain"))
|
|
sv_cz = Statevector.from_instruction(build_preparation_circuit("h_p", "cz_compiled"))
|
|
fid = state_fidelity(sv_cx, sv_cz)
|
|
assert fid == pytest.approx(1.0, abs=1e-6)
|
|
|
|
|
|
def test_encoder_unsupported_raises() -> None:
|
|
with pytest.raises(ValueError, match="Unsupported encoder style"):
|
|
build_encoder("unknown_encoder")
|
|
|
|
|
|
# ── Stabilizer checks for all style combos ───────────────────────────────────
|
|
|
|
@pytest.mark.parametrize("seed,encoder", [
|
|
("h_p", "cx_chain"),
|
|
("h_p", "cz_compiled"),
|
|
("ry_rz", "cx_chain"),
|
|
("ry_rz", "cz_compiled"),
|
|
("u_magic", "cx_chain"),
|
|
("u_magic", "cz_compiled"),
|
|
])
|
|
def test_all_style_combos_satisfy_stabilizers(seed: str, encoder: str) -> None:
|
|
state = Statevector.from_instruction(build_preparation_circuit(seed, encoder))
|
|
for name, stab in STABILIZERS.items():
|
|
exp = state.expectation_value(stab)
|
|
assert abs(exp - 1.0) < 1e-6, (
|
|
f"Stabilizer {name} failed for seed={seed}, encoder={encoder}: exp={exp}"
|
|
)
|