def test_common_backends_compute_2(): resources = get_resources_for_backends( [('numpy', 'cupy'), ( 'numpy', 'cupy', )], user_backends=('numpy', 'cupy'), ) assert resources == {'compute': 1, 'ndarray': 1}
def test_common_backends_as_string(): resources = get_resources_for_backends( ['cupy', ('cupy', )], user_backends='cupy', ) assert resources == {'CUDA': 1, 'compute': 1, 'ndarray': 1}
def test_common_backends_gpu_3(): resources = get_resources_for_backends( [('cupy', ), ('cupy', )], user_backends=('cupy', ), ) assert resources == {'CUDA': 1, 'compute': 1, 'ndarray': 1}
def test_no_common_backends(): with pytest.raises(ValueError): get_resources_for_backends( [('numpy', ), ('cupy', )], user_backends=None, )