Skip to content

Commit

Permalink
Debug test on Windows
Browse files Browse the repository at this point in the history
  • Loading branch information
antonwolfy committed Apr 3, 2024
1 parent 359836a commit 68beec9
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/third_party/cupy/math_tests/test_sumprod.py
Original file line number Diff line number Diff line change
Expand Up @@ -488,14 +488,18 @@ def test_cumsum_numpy_array(self, dtype):
class TestCumprod:
def _cumprod(self, xp, a, *args, **kwargs):
b = a.copy()
print("b.dtype =", b.dtype)
res = xp.cumprod(a, *args, **kwargs)
print("res.dtype =", res.dtype)
testing.assert_array_equal(a, b) # Check if input array is overwritten
return res

@testing.for_all_dtypes()
@testing.numpy_cupy_allclose()
def test_cumprod_1dim(self, xp, dtype):
print(xp)
a = testing.shaped_arange((5,), xp, dtype)
print("a.dtype =", a.dtype)
return self._cumprod(xp, a)

@pytest.mark.usefixtures("allow_fall_back_on_numpy")
Expand Down

0 comments on commit 68beec9

Please sign in to comment.