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")