test gates

Author: pramitsingh0

tests/test_gates.py

@@ -25,13 +25,13 @@ def test_exp_gate():
         ),
         theta=tc.gates.num_to_tensor(np.pi / 2),
     )
-    assert np.allclose(c.wavefunction()[0], -1j)
+    np.testing.assert_allclose(c.wavefunction()[0], -1j)
 
 
 def test_any_gate():
     c = tc.Circuit(2)
     c.any(0, unitary=np.eye(2))
-    assert np.allclose(c.expectation((tc.gates.z(), [0])), 1.0)
+    np.testing.assert_allclose(c.expectation((tc.gates.z(), [0])), 1.0)
 
 
 def test_iswap_gate():