@@ -81,33 +81,38 @@ def test_is_in_device_context_inside_nested_device_ctxt (self):
81
81
class TestIsInDeviceContext (unittest .TestCase ):
82
82
83
83
def test_get_current_device_type_outside_device_ctxt (self ):
84
- self .assertEqual (dpctl .get_current_device_type (), None )
84
+ self .assertNotEqual (dpctl .get_current_device_type (), None )
85
85
86
86
def test_get_current_device_type_inside_device_ctxt (self ):
87
- self .assertEqual (dpctl .get_current_device_type (), None )
87
+ self .assertNotEqual (dpctl .get_current_device_type (), None )
88
88
89
89
with dpctl .device_context ("opencl:gpu:0" ):
90
90
self .assertEqual (
91
91
dpctl .get_current_device_type (), dpctl .device_type .gpu
92
92
)
93
93
94
- self .assertEqual (dpctl .get_current_device_type (), None )
94
+ self .assertNotEqual (dpctl .get_current_device_type (), None )
95
95
96
96
@unittest .skipUnless (
97
97
dpctl .has_cpu_queues (), "No OpenCL CPU queues available"
98
98
)
99
99
def test_get_current_device_type_inside_nested_device_ctxt (self ):
100
- self .assertEqual (dpctl .get_current_device_type (), None )
100
+ self .assertNotEqual (dpctl .get_current_device_type (), None )
101
101
102
102
with dpctl .device_context ("opencl:cpu:0" ):
103
- self .assertEqual (dpctl .get_current_device_type (), dpctl .device_type .cpu )
103
+ self .assertEqual (
104
+ dpctl .get_current_device_type (), dpctl .device_type .cpu
105
+ )
104
106
105
107
with dpctl .device_context ("opencl:gpu:0" ):
106
- self .assertEqual (dpctl .get_current_device_type (), dpctl .
107
- device_type .gpu )
108
- self .assertEqual (dpctl .get_current_device_type (), dpctl .device_type .cpu )
108
+ self .assertEqual (
109
+ dpctl .get_current_device_type (), dpctl .device_type .gpu
110
+ )
111
+ self .assertEqual (
112
+ dpctl .get_current_device_type (), dpctl .device_type .cpu
113
+ )
109
114
110
- self .assertEqual (dpctl .get_current_device_type (), None )
115
+ self .assertNotEqual (dpctl .get_current_device_type (), None )
111
116
112
117
113
118
@unittest .skipIf (not dpctl .has_sycl_platforms (), "No SYCL platforms available" )
0 commit comments