Fix matrix acccess in tests (#100)

* matrix update * matrix update
PennyLaneAI · Jun 7, 2022 · ee8b9a0 · ee8b9a0
1 parent d3eb2f2
commit ee8b9a0
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/tests/test_numpy_wavefunction.py b/tests/test_numpy_wavefunction.py
@@ -113,7 +113,7 @@ def test_apply(self, op, apply_unitary, tol):
         elif op.name == "PSWAP":
             state = apply_unitary(test_operation_map["PSWAP"](0.432), 3)
         else:
-            state = apply_unitary(op.matrix, 3)
+            state = apply_unitary(qml.matrix(op), 3)
 
         with qml.tape.QuantumTape() as tape:
             qml.apply(op)

diff --git a/tests/test_ops.py b/tests/test_ops.py
@@ -22,7 +22,7 @@ def test_cphase_decomposition(self, phi, q, tol):
         calculated_matrix = np.eye(4, dtype=complex)
         default_qubit = qml.device("default.qubit", wires=2)
         for gate in decomposition:
-            gate_matrix = gate.matrix
+            gate_matrix = qml.matrix(gate)
 
             if gate.num_wires == 1:
                 if gate.wires[0] == 0:
@@ -45,7 +45,7 @@ def test_pswap_decomposition(self, phi, tol):
         calculated_matrix = np.eye(4, dtype=complex)
         default_qubit = qml.device("default.qubit", wires=2)
         for gate in decomposition:
-            gate_matrix = gate.matrix
+            gate_matrix = qml.matrix(gate)
 
             if gate.num_wires == 1:
                 if gate.wires[0] == 0:
@@ -67,7 +67,7 @@ def test_iswap_decomposition(self, tol):
         calculated_matrix = np.eye(4, dtype=complex)
         default_qubit = qml.device("default.qubit", wires=2)
         for gate in decomposition:
-            gate_matrix = gate.matrix
+            gate_matrix = qml.matrix(gate)
 
             if gate.num_wires == 1:
                 if gate.wires[0] == 0:

diff --git a/tests/test_qvm.py b/tests/test_qvm.py
@@ -331,7 +331,7 @@ def test_apply(self, op, apply_unitary, shots, qvm, compiler):
         elif op.name == "PSWAP":
             state = apply_unitary(test_operation_map["PSWAP"](0.432), 3)
         else:
-            state = apply_unitary(op.matrix, 3)
+            state = apply_unitary(qml.matrix(op), 3)
 
         with qml.tape.QuantumTape() as tape:
             qml.apply(op)

diff --git a/tests/test_wavefunction.py b/tests/test_wavefunction.py
@@ -113,7 +113,7 @@ def test_apply(self, op, apply_unitary, tol):
         elif op.name == "PSWAP":
             state = apply_unitary(test_operation_map["PSWAP"](0.432), 3)
         else:
-            state = apply_unitary(op.matrix, 3)
+            state = apply_unitary(qml.matrix(op), 3)
 
         with qml.tape.QuantumTape() as tape:
             qml.apply(op)