File size: 2,041 Bytes
2a87ede | 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 | """
Unit tests for dense_evolution/random_circuit.py.
"""
import pytest
from dense_evolution import DenseSVSimulator
from dense_evolution.random_circuit import random_circuit
class TestRandomCircuit:
def test_generates_requested_number_of_gates(self):
assert len(random_circuit(4, 25, seed=1)) == 25
def test_zero_gates_returns_empty_list(self):
assert random_circuit(3, 0, seed=1) == []
def test_reproducible_with_same_seed(self):
a = random_circuit(4, 30, seed=42)
b = random_circuit(4, 30, seed=42)
assert a == b
def test_different_seeds_usually_differ(self):
a = random_circuit(4, 30, seed=1)
b = random_circuit(4, 30, seed=2)
assert a != b
def test_runs_cleanly_on_the_real_simulator(self):
circuit = random_circuit(5, 50, seed=7)
sim = DenseSVSimulator(5, use_gpu=False, use_float32=False)
sim.run_circuit(circuit)
probs = sim.get_probabilities()
assert abs(probs.sum() - 1.0) < 1e-9
def test_single_qubit_never_emits_two_qubit_gates(self):
circuit = random_circuit(1, 30, seed=3)
assert all(len(op) == 2 or op[0] in ('rx', 'ry', 'rz') for op in circuit)
assert all(op[1] == 0 for op in circuit)
def test_gate_set_restricts_output(self):
circuit = random_circuit(4, 40, seed=9, gate_set=['h', 'cx'])
assert all(op[0] in ('h', 'cx') for op in circuit)
def test_unknown_gate_in_gate_set_raises(self):
with pytest.raises(ValueError):
random_circuit(3, 5, gate_set=['not_a_gate'])
def test_rejects_fewer_than_one_qubit(self):
with pytest.raises(ValueError):
random_circuit(0, 5)
def test_rejects_negative_n_gates(self):
with pytest.raises(ValueError):
random_circuit(3, -1)
def test_rejects_out_of_range_two_qubit_prob(self):
with pytest.raises(ValueError):
random_circuit(3, 5, two_qubit_prob=1.5)
|