Add a remote test and fix black

Author: vlad-perevezentsev

dpctl/dptensor/dparray.py

@@ -48,12 +48,13 @@ def dprint(*args):
 def has_array_interface(x):
     return hasattr(x, array_interface_property)
 
+
 def _get_usm_base(ary):
     ob = ary
     while True:
         if ob is None:
             return None
-        elif hasattr(ob, '__sycl_usm_array_interface__'):
+        elif hasattr(ob, "__sycl_usm_array_interface__"):
             return ob
         elif isinstance(ob, np.ndarray):
             ob = ob.base
@@ -92,9 +93,11 @@ def __new__(
                 dprint("buffer None new_obj already has sycl_usm")
             else:
                 dprint("buffer None new_obj will add sycl_usm")
-                setattr(new_obj,
-                        array_interface_property,
-                        new_obj._getter_sycl_usm_array_interface_())
+                setattr(
+                    new_obj,
+                    array_interface_property,
+                    new_obj._getter_sycl_usm_array_interface_(),
+                )
             return new_obj
         # zero copy if buffer is a usm backed array-like thing
         elif hasattr(buffer, array_interface_property):
@@ -113,8 +116,11 @@ def __new__(
                 dprint("buffer None new_obj already has sycl_usm")
             else:
                 dprint("buffer None new_obj will add sycl_usm")
-                setattr(new_obj, array_interface_property,
-                        new_obj._getter_sycl_usm_array_interface_())
+                setattr(
+                    new_obj,
+                    array_interface_property,
+                    new_obj._getter_sycl_usm_array_interface_(),
+                )
             return new_obj
         else:
             dprint("dparray::ndarray __new__ buffer not None and not sycl_usm")
@@ -144,29 +150,30 @@ def __new__(
                 dprint("buffer None new_obj already has sycl_usm")
             else:
                 dprint("buffer None new_obj will add sycl_usm")
-                setattr(new_obj, array_interface_property,
-                        new_obj._getter_sycl_usm_array_interface_())
+                setattr(
+                    new_obj,
+                    array_interface_property,
+                    new_obj._getter_sycl_usm_array_interface_(),
+                )
             return new_obj
 
-
     def _getter_sycl_usm_array_interface_(self):
         ary_iface = self.__array_interface__
         _base = _get_usm_base(self)
         if _base is None:
             raise TypeError
 
-        usm_iface = getattr(_base, '__sycl_usm_array_interface__', None)
+        usm_iface = getattr(_base, "__sycl_usm_array_interface__", None)
         if usm_iface is None:
             raise TypeError
 
-        if (ary_iface['data'][0]  == usm_iface['data'][0]):
-            ary_iface['version'] = usm_iface['version']
-            ary_iface['syclobj'] = usm_iface['syclobj']
+        if ary_iface["data"][0]  == usm_iface["data"][0]:
+            ary_iface["version"] = usm_iface["version"]
+            ary_iface["syclobj"] = usm_iface["syclobj"]
         else:
             raise TypeError
         return ary_iface
 
-
     def __array_finalize__(self, obj):
         dprint("__array_finalize__:", obj, hex(id(obj)), type(obj))
         # When called from the explicit constructor, obj is None
@@ -191,7 +198,6 @@ def __array_finalize__(self, obj):
             "Non-USM allocated ndarray can not viewed as a USM-allocated one without a copy"
         )
 
-
     # Tell Numba to not treat this type just like a NumPy ndarray but to propagate its type.
     # This way it will use the custom dparray allocator.
     __numba_no_subtype_ndarray__ = True

dpctl/tests/test_dparray.py

@@ -52,6 +52,14 @@ def test_multiplication_dparray(self):
         C = self.X * 5
         self.assertIsInstance(C, dparray.ndarray)
 
+    def test_dparray_through_python_func(self):
+        def func_operation_with_const(dpctl_array):
+            return dpctl_array * 2.0 + 13
+
+        C = self.X * 5
+        dp_func = func_operation_with_const(C)
+        self.assertIsInstance(dp_func, dparray.ndarray)
+        
     def test_dparray_mixing_dpctl_and_numpy(self):
         dp_numpy = numpy.ones((256, 4), dtype="d")
         res = dp_numpy * self.X