@@ -68,34 +68,39 @@ def test_greater_equal_complex_matrix(op_dtype):
68
68
sz = 127
69
69
ar1_np_real = np .random .randint (0 , 10 , sz )
70
70
ar1_np_imag = np .random .randint (0 , 10 , sz )
71
- ar1 = dpt .asarray (ar1_np_real + 1j * ar1_np_imag , dtype = op_dtype )
71
+ ar1_np = ar1_np_real + 1j * ar1_np_imag
72
+ ar1 = dpt .asarray (ar1_np , dtype = op_dtype )
72
73
73
74
ar2_np_real = np .random .randint (0 , 10 , sz )
74
75
ar2_np_imag = np .random .randint (0 , 10 , sz )
75
- ar2 = dpt .asarray (ar2_np_real + 1j * ar2_np_imag , dtype = op_dtype )
76
+ ar2_np = ar2_np_real + 1j * ar2_np_imag
77
+ ar2 = dpt .asarray (ar2_np , dtype = op_dtype )
76
78
77
79
r = dpt .greater_equal (ar1 , ar2 )
78
- expected = np .greater_equal (dpt . asnumpy ( ar1 ), dpt . asnumpy ( ar2 ) )
80
+ expected = np .greater_equal (ar1_np , ar2_np )
79
81
assert _compare_dtypes (r .dtype , expected .dtype , sycl_queue = q )
80
82
assert r .shape == expected .shape
81
83
assert (dpt .asnumpy (r ) == expected ).all ()
82
84
83
85
r1 = dpt .greater_equal (ar1 [::- 2 ], ar2 [::2 ])
84
- expected1 = np .greater_equal (dpt . asnumpy ( ar1 [::- 2 ]), dpt . asnumpy ( ar2 [::2 ]) )
86
+ expected1 = np .greater_equal (ar1_np [::- 2 ], ar2_np [::2 ])
85
87
assert _compare_dtypes (r .dtype , expected1 .dtype , sycl_queue = q )
86
88
assert r1 .shape == expected1 .shape
87
89
assert (dpt .asnumpy (r1 ) == expected1 ).all ()
88
90
89
91
ar3 = dpt .asarray ([1.0 + 9j , 2.0 + 0j , 2.0 + 1j , 2.0 + 2j ], dtype = op_dtype )
90
92
ar4 = dpt .asarray ([2.0 + 0j , dpt .nan , dpt .inf , - dpt .inf ], dtype = op_dtype )
93
+
94
+ ar3_np = dpt .asnumpy (ar3 )
95
+ ar4_np = dpt .asnumpy (ar4 )
91
96
r2 = dpt .greater_equal (ar3 , ar4 )
92
97
with np .errstate (invalid = "ignore" ):
93
- expected2 = np .greater_equal (dpt . asnumpy ( ar3 ), dpt . asnumpy ( ar4 ) )
98
+ expected2 = np .greater_equal (ar3_np , ar4_np )
94
99
assert (dpt .asnumpy (r2 ) == expected2 ).all ()
95
100
96
101
r3 = dpt .greater_equal (ar4 , ar4 )
97
102
with np .errstate (invalid = "ignore" ):
98
- expected3 = np .greater_equal (dpt . asnumpy ( ar4 ), dpt . asnumpy ( ar4 ) )
103
+ expected3 = np .greater_equal (ar4_np , ar4_np )
99
104
assert (dpt .asnumpy (r3 ) == expected3 ).all ()
100
105
101
106
@@ -105,23 +110,27 @@ def test_greater_equal_complex_float():
105
110
ar1 = dpt .asarray ([1.0 + 9j , 2.0 + 0j , 2.0 + 1j , 2.0 + 2j ], dtype = "c8" )
106
111
ar2 = dpt .full ((4 ,), 2 , dtype = "f4" )
107
112
113
+ ar1_np = dpt .asnumpy (ar1 )
114
+ ar2_np = dpt .asnumpy (ar2 )
115
+
108
116
r = dpt .greater_equal (ar1 , ar2 )
109
- expected = np .greater_equal (dpt . asnumpy ( ar1 ), dpt . asnumpy ( ar2 ) )
117
+ expected = np .greater_equal (ar1_np , ar2_np )
110
118
assert (dpt .asnumpy (r ) == expected ).all ()
111
119
112
120
r1 = dpt .greater_equal (ar2 , ar1 )
113
- expected1 = np .greater_equal (dpt . asnumpy ( ar2 ), dpt . asnumpy ( ar1 ) )
121
+ expected1 = np .greater_equal (ar2_np , ar1_np )
114
122
assert (dpt .asnumpy (r1 ) == expected1 ).all ()
115
123
with np .errstate (invalid = "ignore" ):
116
124
for tp in [dpt .nan , dpt .inf , - dpt .inf ]:
117
125
118
126
ar3 = dpt .full ((4 ,), tp )
127
+ ar3_np = dpt .asnumpy (ar3 )
119
128
r2 = dpt .greater_equal (ar1 , ar3 )
120
- expected2 = np .greater_equal (dpt . asnumpy ( ar1 ), dpt . asnumpy ( ar3 ) )
129
+ expected2 = np .greater_equal (ar1_np , ar3_np )
121
130
assert (dpt .asnumpy (r2 ) == expected2 ).all ()
122
131
123
132
r3 = dpt .greater_equal (ar3 , ar1 )
124
- expected3 = np .greater_equal (dpt . asnumpy ( ar3 ), dpt . asnumpy ( ar1 ) )
133
+ expected3 = np .greater_equal (ar3_np , ar1_np )
125
134
assert (dpt .asnumpy (r3 ) == expected3 ).all ()
126
135
127
136
0 commit comments