File tree Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Original file line number Diff line number Diff line change @@ -335,6 +335,7 @@ def test_get_component_devices_from_composite():
335335 devices = dpctl .get_devices ()
336336 composite_devices = dpctl .get_composite_devices ()
337337 for Cd in composite_devices :
338- component_devices = Cd .component_devices
338+ component_devices = Cd .component_devices ()
339339 for d in component_devices :
340+ # component devices are root devices
340341 assert d in devices
Original file line number Diff line number Diff line change @@ -259,3 +259,20 @@ def test_platform_get_devices_enum_device_type():
259259 devices = p .get_devices (device_type = dty )
260260 if len (devices ):
261261 assert (d .device_type == dty for d in devices )
262+
263+
264+ def test_platform_get_composite_devices ():
265+ platforms = dpctl .get_platforms ()
266+ if platforms :
267+ for p in platforms :
268+ composite_devices = p .get_composite_devices ()
269+ if not composite_devices :
270+ pass
271+ devices = p .get_devices ()
272+ for Cd in composite_devices :
273+ assert Cd .has_aspect_is_composite
274+ component_devices = Cd .component_devices ()
275+ # all component devices are root devices
276+ assert all (d in devices for d in component_devices )
277+ else :
278+ pytest .skip ("No platforms available" )
You can’t perform that action at this time.
0 commit comments