Update tests/test_states/test_brauer.py

Co-authored-by: Purva Thakre <66048318+purva-thakre@users.noreply.github.com>
vprusso · Nov 15, 2023 · 8179e62 · 8179e62
1 parent 3ea4c1a
commit 8179e62
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_states/test_brauer.py b/tests/test_states/test_brauer.py
@@ -31,5 +31,4 @@
     (2, 2, brauer_2_2),
 ])
 def test_brauer(dim, p_val, expected_result):
-    bool_mat = np.isclose(brauer(dim, p_val), expected_result)
-    np.testing.assert_equal(np.all(bool_mat), True)
+    np.testing.assert_array_equal(brauer(dim, p_val), expected_result)