Skip to content

Commit 05c6b42

Browse files
Merge pull request #1311 from IntelPython/suppress-numpy-warnings-in-test-log
2 parents 22456c4 + f0d1ec9 commit 05c6b42

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

dpctl/tests/elementwise/test_log.py

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,8 @@ def test_log_out_type(dtype):
3131
q = get_queue_or_skip()
3232
skip_if_dtype_not_supported(dtype, q)
3333

34-
X = dpt.asarray(0, dtype=dtype, sycl_queue=q)
35-
expected_dtype = np.log(np.array(0, dtype=dtype)).dtype
34+
X = dpt.asarray(1, dtype=dtype, sycl_queue=q)
35+
expected_dtype = np.log(np.array(1, dtype=dtype)).dtype
3636
expected_dtype = _map_to_device_dtype(expected_dtype, q.sycl_device)
3737
assert dpt.log(X).dtype == expected_dtype
3838

@@ -44,7 +44,7 @@ def test_log_output_contig(dtype):
4444

4545
n_seq = 1027
4646

47-
X = dpt.linspace(0, 13, num=n_seq, dtype=dtype, sycl_queue=q)
47+
X = dpt.linspace(1, 13, num=n_seq, dtype=dtype, sycl_queue=q)
4848
Xnp = dpt.asnumpy(X)
4949

5050
Y = dpt.log(X)
@@ -60,7 +60,7 @@ def test_log_output_strided(dtype):
6060

6161
n_seq = 2 * 1027
6262

63-
X = dpt.linspace(0, 13, num=n_seq, dtype=dtype, sycl_queue=q)[::-2]
63+
X = dpt.linspace(1, 13, num=n_seq, dtype=dtype, sycl_queue=q)[::-2]
6464
Xnp = dpt.asnumpy(X)
6565

6666
Y = dpt.log(X)
@@ -119,11 +119,15 @@ def test_log_special_cases():
119119
q = get_queue_or_skip()
120120

121121
X = dpt.asarray(
122-
[dpt.nan, -1.0, 0.0, -0.0, dpt.inf, -dpt.inf], dtype="f4", sycl_queue=q
122+
[dpt.nan, -dpt.inf, -1.0, -0.0, 0.0, dpt.inf], dtype="f4", sycl_queue=q
123+
)
124+
Y = dpt.log(X)
125+
126+
expected = np.array(
127+
[np.nan, np.nan, np.nan, -np.inf, -np.inf, np.inf], dtype="f4"
123128
)
124-
Xnp = dpt.asnumpy(X)
125129

126-
assert_equal(dpt.asnumpy(dpt.log(X)), np.log(Xnp))
130+
assert_equal(dpt.asnumpy(Y), expected)
127131

128132

129133
@pytest.mark.parametrize("dtype", ["f2", "f4", "f8", "c8", "c16"])

0 commit comments

Comments
 (0)