Skip to content

Commit b1be1ee

Browse files
committed
Add tests for dpctl.SyclPlatform.get_composite_devices
1 parent 1c49cac commit b1be1ee

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

dpctl/tests/test_sycl_device.py

+1
Original file line numberDiff line numberDiff line change
@@ -337,4 +337,5 @@ def test_get_component_devices_from_composite():
337337
for Cd in composite_devices:
338338
component_devices = Cd.component_devices
339339
for d in component_devices:
340+
# component devices are root devices
340341
assert d in devices

dpctl/tests/test_sycl_platform.py

+17
Original file line numberDiff line numberDiff 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+
pytest.skip("No composite devices available")
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")

0 commit comments

Comments
 (0)