Skip to content

Commit e2fdb83

Browse files
vtavanaantonwolfy
andauthored
Apply suggestions from code review
Co-authored-by: Anton <[email protected]>
1 parent d3e8aa3 commit e2fdb83

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dpnp/tests/test_fft.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -929,13 +929,13 @@ def test_rfft_1D_on_2D_array_out(self, dtype, n, axis, norm, order):
929929

930930
@pytest.mark.skipif(not has_support_aspect16(), reason="no fp16 support")
931931
def test_float16(self):
932-
a = numpy.ones(10, dtype=numpy.float16)
932+
a = numpy.arange(10, dtype=numpy.float16)
933933
ia = dpnp.array(a)
934934

935-
result = numpy.fft.rfft(a)
936-
expected = dpnp.fft.rfft(ia)
935+
expected = numpy.fft.rfft(a)
936+
result = dpnp.fft.rfft(ia)
937937
# check_only_type_kind=True since Intel NumPy returns complex128
938-
assert_dtype_allclose(expected, result, check_only_type_kind=True)
938+
assert_dtype_allclose(result, expected, check_only_type_kind=True)
939939

940940
@pytest.mark.parametrize("xp", [numpy, dpnp])
941941
def test_rfft_error(self, xp):

0 commit comments

Comments
 (0)