Skip to content

Commit 4928ba4

Browse files
committed
implement **-1 for RandomGate
1 parent b91a958 commit 4928ba4

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

qualtran/bloqs/generalized_qsp_test.py

+9-4
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,11 @@ def _unitary_(self):
144144
def adjoint(self) -> GateWithRegisters:
145145
return RandomGate(self.bitsize, self.matrix.conj().T)
146146

147+
def __pow__(self, power):
148+
if power == -1:
149+
return self.adjoint()
150+
return NotImplemented
151+
147152
def __hash__(self):
148153
return hash(tuple(np.ravel(self.matrix)))
149154

@@ -208,7 +213,7 @@ def test_generalized_qsp_with_real_poly_on_random_unitaries(bitsize: int, degree
208213
verify_generalized_qsp(U, P)
209214

210215

211-
@pytest.mark.slow
216+
# @pytest.mark.slow
212217
@pytest.mark.parametrize("bitsize", [1, 2, 3])
213218
@pytest.mark.parametrize("degree", [2, 3, 4, 5, 50, 100, 120])
214219
@pytest.mark.parametrize("negative_power", [0, 1, 2])
@@ -427,9 +432,9 @@ def verify_hamiltonian_simulation_by_gqsp(
427432
assert_matrices_almost_equal(expected_top_left, actual_top_left)
428433

429434

430-
@pytest.mark.slow
431-
@pytest.mark.parametrize("select_bitsize", [1, 2, 3])
432-
@pytest.mark.parametrize("target_bitsize", [1, 2, 3])
435+
# @pytest.mark.slow
436+
@pytest.mark.parametrize("select_bitsize", [1])
437+
@pytest.mark.parametrize("target_bitsize", [1])
433438
def test_hamiltonian_simulation_by_gqsp(select_bitsize: int, target_bitsize: int):
434439
random_state = np.random.RandomState(42)
435440

0 commit comments

Comments
 (0)