Skip to content

Commit 9c887e8

Browse files
committed
fix param
1 parent 001ace3 commit 9c887e8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

array_api_strict/tests/test_array_object.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -101,10 +101,10 @@ def test_validate_index():
101101
assert_raises(IndexError, lambda: a[idx])
102102

103103

104-
# @pytest.mark.parametrize("device", ["CPU_DEVICE", "device1", "device2"])
105-
def test_indexing_arrays(device='device1'):
104+
@pytest.mark.parametrize("device", [None, "CPU_DEVICE", "device1", "device2"])
105+
def test_indexing_arrays():
106106
# indexing with 1D integer arrays and mixes of integers and 1D integer are allowed
107-
device = Device(device)
107+
device = None if device is None else Device(device)
108108

109109
# 1D array
110110
a = arange(5)

0 commit comments

Comments
 (0)