From dc78b1d7511dca4297e3fe02340c7e3dae167309 Mon Sep 17 00:00:00 2001
From: Renato Mello <renato.msf@gmail.com>
Date: Tue, 25 Jun 2024 13:22:26 +0400
Subject: [PATCH] tests

---
 tests/test_quantum_info_operations.py | 67 +++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 tests/test_quantum_info_operations.py

diff --git a/tests/test_quantum_info_operations.py b/tests/test_quantum_info_operations.py
new file mode 100644
index 0000000000..7fdd533cb8
--- /dev/null
+++ b/tests/test_quantum_info_operations.py
@@ -0,0 +1,67 @@
+import numpy as np
+import pytest
+
+from qibo import matrices
+from qibo.quantum_info.operations import anticommutator, commutator
+
+
+def test_commutator(backend):
+    matrix_1 = np.random.rand(2, 2, 2)
+    matrix_1 = backend.cast(matrix_1, dtype=matrix_1.dtype)
+
+    matrix_2 = np.random.rand(2, 2)
+    matrix_2 = backend.cast(matrix_2, dtype=matrix_2.dtype)
+
+    with pytest.raises(TypeError):
+        test = commutator(matrix_1, matrix_2)
+    with pytest.raises(TypeError):
+        test = commutator(matrix_2, matrix_1)
+
+    I, X, Y, Z = matrices.I, matrices.X, matrices.Y, matrices.Z
+    I = backend.cast(I, dtype=I.dtype)
+    X = backend.cast(X, dtype=X.dtype)
+    Y = backend.cast(Y, dtype=Y.dtype)
+    Z = backend.cast(Z, dtype=Z.dtype)
+
+    comm = commutator(X, I)
+    backend.assert_allclose(comm, 0.0)
+
+    comm = commutator(X, X)
+    backend.assert_allclose(comm, 0.0)
+
+    comm = commutator(X, Y)
+    backend.assert_allclose(comm, 2j * Z)
+
+    comm = commutator(X, Z)
+    backend.assert_allclose(comm, -2j * Y)
+
+
+def test_anticommutator(backend):
+    matrix_1 = np.random.rand(2, 2, 2)
+    matrix_1 = backend.cast(matrix_1, dtype=matrix_1.dtype)
+
+    matrix_2 = np.random.rand(2, 2)
+    matrix_2 = backend.cast(matrix_2, dtype=matrix_2.dtype)
+
+    with pytest.raises(TypeError):
+        test = anticommutator(matrix_1, matrix_2)
+    with pytest.raises(TypeError):
+        test = anticommutator(matrix_2, matrix_1)
+
+    I, X, Y, Z = matrices.I, matrices.X, matrices.Y, matrices.Z
+    I = backend.cast(I, dtype=I.dtype)
+    X = backend.cast(X, dtype=X.dtype)
+    Y = backend.cast(Y, dtype=Y.dtype)
+    Z = backend.cast(Z, dtype=Z.dtype)
+
+    anticomm = anticommutator(X, I)
+    backend.assert_allclose(anticomm, 2 * X)
+
+    anticomm = anticommutator(X, X)
+    backend.assert_allclose(anticomm, 2 * I)
+
+    anticomm = anticommutator(X, Y)
+    backend.assert_allclose(anticomm, 0.0)
+
+    anticomm = anticommutator(X, Z)
+    backend.assert_allclose(anticomm, 0.0)