@@ -44,16 +44,16 @@ def test_isin_basic(dtype):
44
44
skip_if_dtype_not_supported (dtype , q )
45
45
46
46
n = 100
47
- x = dpt .arange (n , dtype = dtype )
48
- test = dpt .arange (n - 1 , dtype = dtype )
47
+ x = dpt .arange (n , dtype = dtype , sycl_queue = q )
48
+ test = dpt .arange (n - 1 , dtype = dtype , sycl_queue = q )
49
49
r1 = dpt .isin (x , test )
50
50
assert dpt .all (r1 [:- 1 ])
51
51
assert not r1 [- 1 ]
52
52
assert r1 .shape == x .shape
53
53
54
54
# test with invert keyword
55
55
r2 = dpt .isin (x , test , invert = True )
56
- assert not dpt .all (r2 [:- 1 ])
56
+ assert not dpt .any (r2 [:- 1 ])
57
57
assert r2 [- 1 ]
58
58
assert r2 .shape == x .shape
59
59
@@ -70,7 +70,7 @@ def test_isin_basic_bool():
70
70
assert r1 .shape == x .shape
71
71
72
72
r2 = dpt .isin (x , test , invert = True )
73
- assert not dpt .all (r2 [:- 1 ])
73
+ assert not dpt .any (r2 [:- 1 ])
74
74
assert r2 [- 1 ]
75
75
assert r2 .shape == x .shape
76
76
@@ -98,37 +98,44 @@ def test_isin_strided(dtype):
98
98
skip_if_dtype_not_supported (dtype , q )
99
99
100
100
n , m = 100 , 20
101
- x = dpt .zeros ((n , m ), dtype = dtype , order = "F" )
102
- x [:, ::2 ] = dpt .arange (1 , (m / 2 ) + 1 , dtype = dtype )
103
- test = dpt .arange (1 , (m / 2 ) + 1 , dtype = dtype )
104
- r1 = dpt .isin (x , test )
105
- assert dpt .all (r1 [:, ::2 ])
106
- assert not dpt .all (r1 [:, 1 ::2 ])
107
- assert r1 .shape == x .shape
101
+ x = dpt .zeros ((n , m ), dtype = dtype , order = "F" , sycl_queue = q )
102
+ x [:, ::2 ] = dpt .arange (1 , (m / 2 ) + 1 , dtype = dtype , sycl_queue = q )
103
+ x_s = x [:, ::2 ]
104
+ test = dpt .arange (1 , (m / 2 ), dtype = dtype , sycl_queue = q )
105
+ r1 = dpt .isin (x_s , test )
106
+ assert dpt .all (r1 [:, :- 1 ])
107
+ assert not dpt .any (r1 [:, - 1 ])
108
+ assert not dpt .any (x [:, 1 ::2 ])
109
+ assert r1 .shape == x_s .shape
108
110
109
111
# test with invert keyword
110
- r2 = dpt .isin (x , test , invert = True )
111
- assert not dpt .all (r2 [:, ::2 ])
112
- assert dpt .all (r2 [:, 1 ::2 ])
113
- assert r2 .shape == x .shape
112
+ r2 = dpt .isin (x_s , test , invert = True )
113
+ assert not dpt .any (r2 [:, :- 1 ])
114
+ assert dpt .all (r2 [:, - 1 ])
115
+ assert not dpt .any (x [:, 1 :2 ])
116
+ assert r2 .shape == x_s .shape
114
117
115
118
116
119
def test_isin_strided_bool ():
117
120
dt = dpt .bool
121
+
118
122
n , m = 100 , 20
119
- x = dpt .ones ((n , m ), dtype = dt , order = "F" )
120
- x [:, ::2 ] = False
121
- test = dpt .zeros ((), dtype = dt )
122
- r1 = dpt .isin (x , test )
123
- assert dpt .all (r1 [:, ::2 ])
124
- assert not dpt .all (r1 [:, 1 ::2 ])
125
- assert r1 .shape == x .shape
123
+ x = dpt .zeros ((n , m ), dtype = dt , order = "F" )
124
+ x [:, :- 2 :2 ] = True
125
+ x_s = x [:, ::2 ]
126
+ test = dpt .ones ((), dtype = dt )
127
+ r1 = dpt .isin (x_s , test )
128
+ assert dpt .all (r1 [:, :- 1 ])
129
+ assert not dpt .any (r1 [:, - 1 ])
130
+ assert not dpt .any (x [:, 1 ::2 ])
131
+ assert r1 .shape == x_s .shape
126
132
127
133
# test with invert keyword
128
- r2 = dpt .isin (x , test , invert = True )
129
- assert not dpt .all (r2 [:, ::2 ])
130
- assert dpt .all (r2 [:, 1 ::2 ])
131
- assert r2 .shape == x .shape
134
+ r2 = dpt .isin (x_s , test , invert = True )
135
+ assert not dpt .any (r2 [:, :- 1 ])
136
+ assert dpt .all (r2 [:, - 1 ])
137
+ assert not dpt .any (x [:, 1 :2 ])
138
+ assert r2 .shape == x_s .shape
132
139
133
140
134
141
def test_isin_empty_inputs ():
0 commit comments