Skip to content

Commit

Permalink
test: relax more assertion in test_bilinear
Browse files Browse the repository at this point in the history
  • Loading branch information
mrava87 committed Sep 13, 2024
1 parent efb2839 commit 1f48af5
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions pytests/test_bilinear.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,8 @@ def test_lrfactorized(par):
X = U @ V.T
LOp = LowRankFactorizedMatrix(U, V.T, X)

assert_array_almost_equal(X.ravel(), LOp._matvecx(U.ravel()))
assert_array_almost_equal(X.ravel(), LOp._matvecy(V.T.ravel()))
assert_array_almost_equal(X.ravel(), LOp._matvecx(U.ravel()), decimal=4)
assert_array_almost_equal(X.ravel(), LOp._matvecy(V.T.ravel()), decimal=4)


@pytest.mark.parametrize("par", [(par1), (par2)])
Expand All @@ -46,5 +46,5 @@ def test_lrfactorizedoperator(par):

LOp = LowRankFactorizedMatrix(U, V.T, y, Op)

assert_array_almost_equal(y, LOp._matvecx(U.ravel()))
assert_array_almost_equal(y, LOp._matvecy(V.T.ravel()))
assert_array_almost_equal(y, LOp._matvecx(U.ravel()), decimal=4)
assert_array_almost_equal(y, LOp._matvecy(V.T.ravel()), decimal=4)

0 comments on commit 1f48af5

Please sign in to comment.