def test_expand_requested_device_ids_exception(requested_device_ids, num_gpus_available): with pytest.raises(ValueError): utils._expand_requested_device_ids(requested_device_ids, num_gpus_available)
def test_expand_requested_device_ids(requested_device_ids, num_gpus_available, expected): assert set( utils._expand_requested_device_ids( requested_device_ids, num_gpus_available)) == set(expected)