def test_matrix_elem(self, wires, vec1, obs, vec2, expected):
"""Tests for the helper function _matrix_elem"""
dev = qml.device("default.qubit", wires=wires)
tape = ReversibleTape()
res = tape._matrix_elem(vec1, obs, vec2, dev)
assert res == expected
def test_matrix_elem(self, wires, vec1, obs, vec2, expected):
"""Tests for the helper function _matrix_elem"""
tape = ReversibleTape()
res = tape._matrix_elem(vec1, obs, vec2, qml.wires.Wires(range(wires)))
assert res == expected
