Exemple #1
0
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)
Exemple #2
0
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)