def test_remove_unused_qubits_2(self):
"""
Tests that unused qubits are found correctly.
"""
operator = (I ^ I ^ I ^ Z ^ I) + (I ^ I ^ Z ^ I ^ I)
reduced, unused = _remove_unused_qubits(operator)
expected_reduced = 1.0 * (I ^ Z) + 1.0 * (Z ^ I)
expected_unused = [0, 3, 4]
self.assertEqual(reduced, expected_reduced)
self.assertEqual(unused, expected_unused)
def test_remove_unused_qubits_pauli_op_2(self):
"""
Tests that unused qubits are found correctly.
"""
operator = I ^ Z ^ I ^ Z ^ I
reduced, unused = _remove_unused_qubits(operator)
expected_reduced = 1.0 * (Z ^ Z)
expected_unused = [0, 2, 4]
self.assertEqual(reduced, expected_reduced)
self.assertEqual(unused, expected_unused)
