def test_bit_driver_output(self):
"""Tests that the bit driver Hamiltonian has the correct output"""
H = qaoa.bit_driver(range(3), 1)
hamiltonian = qml.Hamiltonian([1, 1, 1], [qml.PauliZ(0), qml.PauliZ(1), qml.PauliZ(2)])
assert decompose_hamiltonian(H) == decompose_hamiltonian(hamiltonian)
def test_bit_driver_error(self):
"""Tests that the bit driver Hamiltonian throws the correct error"""
with pytest.raises(ValueError, match=r"'b' must be either 0 or 1"):
qaoa.bit_driver(range(3), 2)
