提交 2c9ee770 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Rename test_numba_* tests in tests.link.test_numba

上级 c8908e5a
...@@ -309,7 +309,7 @@ def test_create_numba_signature(v, expected, force_scalar): ...@@ -309,7 +309,7 @@ def test_create_numba_signature(v, expected, force_scalar):
) )
], ],
) )
def test_numba_box_unbox(input, wrapper_fn, check_fn): def test_box_unbox(input, wrapper_fn, check_fn):
input = wrapper_fn(input) input = wrapper_fn(input)
pass_through = numba.njit(lambda x: x) pass_through = numba.njit(lambda x: x)
...@@ -419,7 +419,7 @@ def test_Elemwise(inputs, input_vals, output_fn, exc): ...@@ -419,7 +419,7 @@ def test_Elemwise(inputs, input_vals, output_fn, exc):
), ),
], ],
) )
def test_numba_Composite(inputs, input_values, scalar_fn): def test_Composite(inputs, input_values, scalar_fn):
composite_inputs = [aes.float64(i.name) for i in inputs] composite_inputs = [aes.float64(i.name) for i in inputs]
comp_op = Elemwise(Composite(composite_inputs, [scalar_fn(*composite_inputs)])) comp_op = Elemwise(Composite(composite_inputs, [scalar_fn(*composite_inputs)]))
out_fg = FunctionGraph(inputs, [comp_op(*inputs)]) out_fg = FunctionGraph(inputs, [comp_op(*inputs)])
...@@ -3530,7 +3530,7 @@ def test_scan_while(): ...@@ -3530,7 +3530,7 @@ def test_scan_while():
), ),
], ],
) )
def test_numba_ifelse(inputs, cond_fn, true_vals, false_vals): def test_IfElse(inputs, cond_fn, true_vals, false_vals):
out = ifelse(cond_fn(*inputs), true_vals, false_vals) out = ifelse(cond_fn(*inputs), true_vals, false_vals)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论