def test_common_backends_string(default_raw, lt_ctx):
runner = UDFRunner([UDF4(), UDF5()])
tasks = list(
runner._make_udf_tasks(
dataset=default_raw,
roi=None,
backends=None,
))
for task in tasks:
assert task.get_resources() == {'CUDA': 1, 'compute': 1}
def test_common_backends_cpu(default_raw, lt_ctx):
runner = UDFRunner([UDF1(), UDF3()])
tasks = list(
runner._make_udf_tasks(
dataset=default_raw,
roi=None,
backends=None,
))
for task in tasks:
assert task.get_resources() == {'CPU': 1, 'compute': 1, 'ndarray': 1}
def test_no_common_backends(default_raw, lt_ctx):
runner = UDFRunner([UDF1(), UDF2()])
tasks = list(
runner._make_udf_tasks(
dataset=default_raw,
roi=None,
backends=None,
))
for task in tasks:
with pytest.raises(ValueError) as e:
task.get_resources()
assert e.match("^There is no common supported UDF backend")