diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-26 11:47:29 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-26 11:47:29 +0200 |
commit | 4a45135dfa3f1aeae8b3bdf0c42422f0f60426e8 (patch) | |
tree | da3d2b6f629e2b377ff0aedd6405bcaac9baefda /test/test_dr.py | |
parent | 347e6288b87cbeef9b8fbc1a08cd130b96de1d61 (diff) |
dr +gpu numpy assert
Diffstat (limited to 'test/test_dr.py')
-rw-r--r-- | test/test_dr.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_dr.py b/test/test_dr.py index e3d1e6b..bdb920e 100644 --- a/test/test_dr.py +++ b/test/test_dr.py @@ -29,7 +29,7 @@ def test_fda(): projfda(xs) - assert np.allclose(np.sum(Pfda**2, 0), np.ones(p)) + np.testing.assert_allclose(np.sum(Pfda**2, 0), np.ones(p)) @pytest.mark.skipif(nogo, reason="Missing modules (autograd or pymanopt)") @@ -51,4 +51,4 @@ def test_wda(): projwda(xs) - assert np.allclose(np.sum(Pwda**2, 0), np.ones(p)) + np.testing.assert_allclose(np.sum(Pwda**2, 0), np.ones(p)) |