From 805e4428cd8198a4259b0808b3790685a8c3f4e6 Mon Sep 17 00:00:00 2001 From: madcpf Date: Sat, 20 Jul 2024 00:05:49 -0700 Subject: [PATCH] fix test --- unitary/alpha/quantum_world_test.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/unitary/alpha/quantum_world_test.py b/unitary/alpha/quantum_world_test.py index c7f61a69..d168c077 100644 --- a/unitary/alpha/quantum_world_test.py +++ b/unitary/alpha/quantum_world_test.py @@ -905,9 +905,7 @@ def test_density_matrix(): testing.assert_array_equal(board.density_matrix(objects=[light1]), rho_green) testing.assert_array_equal(board.density_matrix(objects=[light2]), rho_red) - testing.assert_array_equal( - board.density_matrix(objects=[light3]), rho_red, rho_green - ) + testing.assert_array_equal(board.density_matrix(objects=[light3]), rho_red) testing.assert_array_equal( board.density_matrix(objects=[light1, light2]), np.kron(rho_green, rho_red) @@ -916,7 +914,7 @@ def test_density_matrix(): board.density_matrix(objects=[light2, light3]), np.kron(rho_red, rho_red) ) testing.assert_array_equal( - board.density_matrix(objects=[light1, light3]), np.kron(rho_green, rho_red) + board.density_matrix(objects=[light3, light1]), np.kron(rho_red, rho_green) ) testing.assert_array_equal(