Skip to content

Commit 5f681bc

Browse files
Add more tests
1 parent 94dccbc commit 5f681bc

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

tests/test_logic.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -431,6 +431,34 @@ def test_finite(op, data, dtype):
431431
assert_equal(dpnp_res, np_res)
432432

433433

434+
class TestIsFortran:
435+
@pytest.mark.parametrize(
436+
"array, expected",
437+
[
438+
(dpnp.ones((2, 4), order="C"), True),
439+
(dpnp.ones((2, 4), order="F"), False),
440+
],
441+
)
442+
def test_isfortran_transpose(self, array, expected):
443+
assert dpnp.isfortran(array.T) == expected
444+
445+
@pytest.mark.parametrize(
446+
"array, expected",
447+
[
448+
(dpnp.ones((2, 4), order="C"), False),
449+
(dpnp.ones((2, 4), order="F"), True),
450+
],
451+
)
452+
def test_isfortran_usm_ndarray(self, array, expected):
453+
assert dpnp.isfortran(array.get_array()) == expected
454+
455+
def test_isfortran_errors(self):
456+
# unsupported type
457+
a_np = numpy.ones((2, 3))
458+
assert_raises(TypeError, dpnp.isfortran, a_np)
459+
assert_raises(TypeError, dpnp.isfortran, [1, 2, 3])
460+
461+
434462
@pytest.mark.parametrize("func", ["isneginf", "isposinf"])
435463
@pytest.mark.parametrize(
436464
"data",

0 commit comments

Comments
 (0)