added memory.py to expose dpclt.memory module

modularized test, + changes per black

Author: oleksandr-pavlyk

dpctl/memory.py

@@ -0,0 +1,3 @@
+from ._memory import MemoryUSMShared, MemoryUSMDevice, MemoryUSMHost
+
+__all__ = ["MemoryUSMShared", "MemoryUSMDevice", "MemoryUSMHost"]

dpctl/tests/test_sycl_usm.py

@@ -36,14 +36,20 @@ def test_memory_create(self):
         queue = dpctl.get_current_queue()
         mobj = MemoryUSMShared(nbytes, queue)
         self.assertEqual(mobj.nbytes, nbytes)
-        self.assertTrue(hasattr(mobj, '__sycl_usm_array_interface__'))
+        self.assertTrue(hasattr(mobj, "__sycl_usm_array_interface__"))
 
     def _create_memory(self):
         nbytes = 1024
         queue = dpctl.get_current_queue()
         mobj = MemoryUSMShared(nbytes, queue)
         return mobj
 
+    def _create_host_buf(self, nbytes):
+        ba = bytearray(nbytes)
+        for i in range(nbytes):
+            ba[i] = (i % 32) + ord("a")
+        return ba
+
     @unittest.skipUnless(
         dpctl.has_sycl_platforms(), "No SYCL devices except the default host device."
     )
@@ -97,9 +103,7 @@ def test_buffer_protocol(self):
     )
     def test_copy_host_roundtrip(self):
         mobj = self._create_memory()
-        host_src_obj = bytearray(mobj.nbytes)
-        for i in range(mobj.nbytes):
-            host_src_obj[i] = (i % 32) + ord('a')
+        host_src_obj = self._create_host_buf(mobj.nbytes)
         mobj.copy_from_host(host_src_obj)
         host_dest_obj = mobj.copy_to_host()
         del mobj
@@ -113,22 +117,24 @@ def test_zero_copy(self):
         mobj2 = type(mobj)(mobj)
 
         self.assertTrue(mobj2.reference_obj is mobj)
-        self.assertTrue(mobj2.__sycl_usm_array_interface__['data'] == mobj.__sycl_usm_array_interface__['data'])
+        mobj_data = mobj.__sycl_usm_array_interface__["data"]
+        mobj2_data = mobj2.__sycl_usm_array_interface__["data"]
+        self.assertEqual(mobj_data, mobj2_data)
 
     @unittest.skipUnless(
         dpctl.has_sycl_platforms(), "No SYCL devices except the default host device."
     )
     def test_pickling(self):
         import pickle
+
         mobj = self._create_memory()
-        host_src_obj = bytearray(mobj.nbytes)
-        for i in range(mobj.nbytes):
-            host_src_obj[i] = (i % 32) + ord('a')
+        host_src_obj = self._create_host_buf(mobj.nbytes)
         mobj.copy_from_host(host_src_obj)
 
-        mobj2 = pickle.loads(pickle.dumps(mobj))
-        self.assertEqual(mobj.tobytes(), mobj2.tobytes())
-        self.assertNotEqual(mobj._pointer, mobj2._pointer)        
+        mobj_reconstructed = pickle.loads(pickle.dumps(mobj))
+        self.assertEqual(mobj.tobytes(), mobj_reconstructed.tobytes())
+        self.assertNotEqual(mobj._pointer, mobj_reconstructed._pointer)
+
 
 class TestMemoryUSMBase:
     """ Base tests for MemoryUSM* """
@@ -175,7 +181,5 @@ class TestMemoryUSMDevice(TestMemoryUSMBase, unittest.TestCase):
     usm_type = "device"
 
 
-    
-
 if __name__ == "__main__":
     unittest.main()