def test_domino_8(self):
"""Domino with index = 8."""
e_0, e_1, e_2 = ket(3, 0), ket(3, 1), ket(3, 2)
expected_res = np.kron(1 / np.sqrt(2) * (e_0 - e_1), e_2)
res = domino(8)
bool_mat = np.isclose(res, expected_res)
self.assertEqual(np.all(bool_mat), True)
def test_domino_2(self):
"""Domino with index = 2."""
e_0, e_1 = ket(3, 0), ket(3, 1)
expected_res = np.kron(e_0, 1 / np.sqrt(2) * (e_0 - e_1))
res = domino(2)
bool_mat = np.isclose(res, expected_res)
self.assertEqual(np.all(bool_mat), True)
def test_domino_4(self):
"""Domino with index = 4."""
e_1, e_2 = ket(3, 1), ket(3, 2)
expected_res = np.kron(e_2, 1 / np.sqrt(2) * (e_1 - e_2))
res = domino(4)
bool_mat = np.isclose(res, expected_res)
self.assertEqual(np.all(bool_mat), True)
def test_domino_0(self):
"""Domino with index = 0."""
e_1 = ket(3, 1)
expected_res = np.kron(e_1, e_1)
res = domino(0)
bool_mat = np.isclose(res, expected_res)
self.assertEqual(np.all(bool_mat), True)
def test_invalid_index(self):
"""Tests for invalid index input."""
with self.assertRaises(ValueError):
domino(9)