Update tests/test_states/test_basis.py

Co-authored-by: Purva Thakre <66048318+purva-thakre@users.noreply.github.com>
vprusso · Nov 15, 2023 · 5512830 · 5512830
1 parent 7acae82
commit 5512830
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_states/test_basis.py b/tests/test_states/test_basis.py
@@ -14,8 +14,7 @@
     (4, 0, np.array([[1], [0], [0], [0]])),
 ])
 def test_basis(dim, pos, expected_result):
-    bool_mat = np.isclose(basis(dim, pos), expected_result)
-    np.testing.assert_equal(np.all(bool_mat), True)
+    np.testing.assert_array_equal(basis(dim, pos), expected_result)
 
 
 def test_basis_invalid_dim():