@@ -164,11 +164,11 @@ def test_pickling(self):
164164 )
165165
166166
167- class TestMemoryUSMBase :
167+ class _TestMemoryUSMBase :
168168 """ Base tests for MemoryUSM* """
169169
170- MemoryUSMClass = None
171- usm_type = None
170+ def setUp ( self ):
171+ pass
172172
173173 @unittest .skipUnless (
174174 dpctl .has_sycl_platforms (), "No SYCL devices except the default host device."
@@ -217,25 +217,28 @@ def test_sycl_usm_array_interface(self):
217217 self .assertTrue (np .array_equal (m .copy_to_host (), hb ))
218218
219219
220- class TestMemoryUSMShared (TestMemoryUSMBase , unittest .TestCase ):
220+ class TestMemoryUSMShared (_TestMemoryUSMBase , unittest .TestCase ):
221221 """ Tests for MemoryUSMShared """
222222
223- MemoryUSMClass = MemoryUSMShared
224- usm_type = "shared"
223+ def setUp (self ):
224+ self .MemoryUSMClass = MemoryUSMShared
225+ self .usm_type = "shared"
225226
226227
227- class TestMemoryUSMHost (TestMemoryUSMBase , unittest .TestCase ):
228+ class TestMemoryUSMHost (_TestMemoryUSMBase , unittest .TestCase ):
228229 """ Tests for MemoryUSMHost """
229230
230- MemoryUSMClass = MemoryUSMHost
231- usm_type = "host"
231+ def setUp (self ):
232+ self .MemoryUSMClass = MemoryUSMHost
233+ self .usm_type = "host"
232234
233235
234- class TestMemoryUSMDevice (TestMemoryUSMBase , unittest .TestCase ):
236+ class TestMemoryUSMDevice (_TestMemoryUSMBase , unittest .TestCase ):
235237 """ Tests for MemoryUSMDevice """
236238
237- MemoryUSMClass = MemoryUSMDevice
238- usm_type = "device"
239+ def setUp (self ):
240+ self .MemoryUSMClass = MemoryUSMDevice
241+ self .usm_type = "device"
239242
240243
241244if __name__ == "__main__" :
0 commit comments