Skip to content

Commit f1bf82b

Browse files
committed
Fix the test input
1 parent fd13907 commit f1bf82b

File tree

1 file changed

+4
-21
lines changed

1 file changed

+4
-21
lines changed

tests/test_manipulation.py

Lines changed: 4 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -333,27 +333,10 @@ def test_no_copy(self):
333333

334334

335335
class TestBroadcast:
336-
@pytest.mark.parametrize(
337-
"shapes",
338-
[
339-
[[(1,), (3,)]],
340-
[[(1, 3), (3, 3)]],
341-
[[(3, 1), (3, 3)]],
342-
[[(1, 3), (3, 1)]],
343-
[[(1, 1), (3, 3)]],
344-
[[(1, 1), (1, 3)]],
345-
[[(1, 1), (3, 1)]],
346-
[[(1, 0), (0, 0)]],
347-
[[(0, 1), (0, 0)]],
348-
[[(1, 0), (0, 1)]],
349-
[[(1, 1), (0, 0)]],
350-
[[(1, 1), (1, 0)]],
351-
[[(1, 1), (0, 1)]],
352-
],
353-
)
354-
def test_broadcast_shapes(self, shapes):
355-
expected = numpy.broadcast_shapes(*shapes)
356-
result = dpnp.broadcast_shapes(*shapes)
336+
@pytest.mark.parametrize("shape", [(1, 1), (0, 1)])
337+
def test_broadcast_shapes(self, shape):
338+
expected = numpy.broadcast_shapes(*shape)
339+
result = dpnp.broadcast_shapes(*shape)
357340
assert_equal(result, expected)
358341

359342

0 commit comments

Comments
 (0)