Ejemplo n.º 1
0
def test_parse_gpu_returns_none_when_no_devices_are_available(
        mocked_device_count_0, gpus):
    with pytest.raises(MisconfigurationException):
        _parse_gpu_ids(gpus)
Ejemplo n.º 2
0
def test_parse_gpu_fail_on_non_existent_id_2(mocked_device_count):
    with pytest.raises(MisconfigurationException):
        _parse_gpu_ids([1, 2, 19])
Ejemplo n.º 3
0
def test_parse_gpu_fail_on_unsupported_inputs(mocked_device_count, gpus):
    with pytest.raises(MisconfigurationException):
        _parse_gpu_ids(gpus)
Ejemplo n.º 4
0
def test_parse_gpu_fail_on_non_existent_id(mocked_device_count_0, gpus):
    with pytest.raises(MisconfigurationException):
        _parse_gpu_ids(gpus)
Ejemplo n.º 5
0
def test_parse_gpu_ids(mocked_device_count, gpus, expected_gpu_ids):
    assert _parse_gpu_ids(gpus) == expected_gpu_ids