Select correct devices when tests run with XOBJECTS_TEST_CONTEXTS

Author: szymonlopaciuk

xobjects/context_cupy.py

@@ -473,6 +473,9 @@ def build_kernels(
 
         return out_kernels
 
+    def __str__(self):
+        return f"{type(self).__name__}:{cupy.cuda.get_device_id()}"
+
     def nparray_to_context_array(self, arr):
         """
         Copies a numpy array to the device memory.

xobjects/context_pyopencl.py

@@ -219,7 +219,7 @@ def build_kernels(
                 fid.write(specialized_source)
 
         prg = cl.Program(self.context, specialized_source).build(
-            options="-cl-std=CL2.0"
+            options="-cl-std=CL2.0",
         )
 
         out_kernels = {}
@@ -238,6 +238,11 @@ def build_kernels(
 
         return out_kernels
 
+    def __str__(self):
+        platform_id = cl.get_platforms().index(self.platform)
+        device_id = self.platform.get_devices().index(self.device)
+        return f"{type(self).__name__}:{platform_id}.{device_id}"
+
     def nparray_to_context_array(self, arr):
         """
         Copies a numpy array to the device memory.