Skip to content

Commit 2c622fa

Browse files
Fix/test cases with pytest (#275)
* fixed test_sycl_usm to with both pytest and unittests Co-authored-by: Oleksandr Pavlyk <[email protected]>
1 parent 11b969b commit 2c622fa

File tree

1 file changed

+15
-12
lines changed

1 file changed

+15
-12
lines changed

dpctl/tests/test_sycl_usm.py

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

241244
if __name__ == "__main__":

0 commit comments

Comments
 (0)