diff options
Diffstat (limited to 'test/test_ot.py')
-rw-r--r-- | test/test_ot.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_ot.py b/test/test_ot.py index ac86602..dacae0a 100644 --- a/test/test_ot.py +++ b/test/test_ot.py @@ -72,7 +72,8 @@ def test_emd_1d_emd2_1d(): # check AssertionError is raised if called on non 1d arrays u = np.random.randn(n, 2) v = np.random.randn(m, 2) - np.testing.assert_raises(AssertionError, ot.emd_1d, u, v, [], []) + with pytest.raises(AssertionError): + ot.emd_1d(u, v, [], []) def test_wass_1d(): |