def test_is_not_unitary_matrix(self):
"""Test that non-unitary matrix returns False."""
mat = np.array([[1, 0], [1, 1]])
mat = np.matrix(mat)
self.assertEqual(is_unitary(mat), False)
def test_is_unitary_hardcoded(self):
"""Test that hardcoded unitary matrix returns True."""
mat = np.array([[0, 1], [1, 0]])
self.assertEqual(is_unitary(mat), True)
def test_random_unitary_not_real(self):
"""Generate random non-real unitary matrix."""
mat = random_unitary(2)
self.assertEqual(is_unitary(mat), True)
def test_is_unitary_random(self):
"""Test that random unitary matrix returns True."""
mat = random_unitary(2)
self.assertEqual(is_unitary(mat), True)
def test_random_unitary_vec_dim(self):
"""Generate random non-real unitary matrix."""
mat = random_unitary([4, 4], True)
self.assertEqual(is_unitary(mat), True)