Skip to content

Commit 15e568a

Browse files
committed
Add testing of tuple to test_meshgrid
1 parent 55df20c commit 15e568a

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

dpnp/tests/test_arraycreation.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -984,13 +984,19 @@ def test_dpctl_tensor_input(func, args):
984984
[[], [[1]], [[1, 2, 3], [4, 5, 6]], [[1, 2], [3, 4], [5, 6]]],
985985
ids=["[]", "[[1]]", "[[1, 2, 3], [4, 5, 6]]", "[[1, 2], [3, 4], [5, 6]]"],
986986
)
987-
@pytest.mark.parametrize("dtype", get_all_dtypes(no_float16=False))
987+
@pytest.mark.parametrize(
988+
"dtype", get_all_dtypes(no_none=True, no_float16=False)
989+
)
988990
@pytest.mark.parametrize("indexing", ["ij", "xy"])
989991
def test_meshgrid(arrays, dtype, indexing):
990992
func = lambda xp, xi: xp.meshgrid(*xi, indexing=indexing)
991993
a = tuple(numpy.array(array, dtype=dtype) for array in arrays)
992994
ia = tuple(dpnp.array(array, dtype=dtype) for array in arrays)
993-
assert_array_equal(func(numpy, a), func(dpnp, ia))
995+
996+
result = func(dpnp, ia)
997+
expected = func(numpy, a)
998+
assert_array_equal(result, expected, strict=True)
999+
assert isinstance(result, tuple)
9941000

9951001

9961002
@pytest.mark.parametrize("shape", [(24,), (4, 6), (2, 3, 4), (2, 3, 2, 2)])

0 commit comments

Comments
 (0)