@@ -164,11 +164,11 @@ def test_pickling(self):
164
164
)
165
165
166
166
167
- class TestMemoryUSMBase :
167
+ class _TestMemoryUSMBase :
168
168
""" Base tests for MemoryUSM* """
169
169
170
- MemoryUSMClass = None
171
- usm_type = None
170
+ def setUp ( self ):
171
+ pass
172
172
173
173
@unittest .skipUnless (
174
174
dpctl .has_sycl_platforms (), "No SYCL devices except the default host device."
@@ -217,25 +217,28 @@ def test_sycl_usm_array_interface(self):
217
217
self .assertTrue (np .array_equal (m .copy_to_host (), hb ))
218
218
219
219
220
- class TestMemoryUSMShared (TestMemoryUSMBase , unittest .TestCase ):
220
+ class TestMemoryUSMShared (_TestMemoryUSMBase , unittest .TestCase ):
221
221
""" Tests for MemoryUSMShared """
222
222
223
- MemoryUSMClass = MemoryUSMShared
224
- usm_type = "shared"
223
+ def setUp (self ):
224
+ self .MemoryUSMClass = MemoryUSMShared
225
+ self .usm_type = "shared"
225
226
226
227
227
- class TestMemoryUSMHost (TestMemoryUSMBase , unittest .TestCase ):
228
+ class TestMemoryUSMHost (_TestMemoryUSMBase , unittest .TestCase ):
228
229
""" Tests for MemoryUSMHost """
229
230
230
- MemoryUSMClass = MemoryUSMHost
231
- usm_type = "host"
231
+ def setUp (self ):
232
+ self .MemoryUSMClass = MemoryUSMHost
233
+ self .usm_type = "host"
232
234
233
235
234
- class TestMemoryUSMDevice (TestMemoryUSMBase , unittest .TestCase ):
236
+ class TestMemoryUSMDevice (_TestMemoryUSMBase , unittest .TestCase ):
235
237
""" Tests for MemoryUSMDevice """
236
238
237
- MemoryUSMClass = MemoryUSMDevice
238
- usm_type = "device"
239
+ def setUp (self ):
240
+ self .MemoryUSMClass = MemoryUSMDevice
241
+ self .usm_type = "device"
239
242
240
243
241
244
if __name__ == "__main__" :
0 commit comments