提交 9874a855 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Do not compile test functions with random constants

This way they can be fully cached when re-running tests
上级 6ac5ab28
...@@ -9,7 +9,7 @@ from tests.link.numba.test_basic import compare_numba_and_py ...@@ -9,7 +9,7 @@ from tests.link.numba.test_basic import compare_numba_and_py
@pytest.mark.parametrize( @pytest.mark.parametrize(
"x", "x_test",
[ [
[], # Empty list [], # Empty list
[3, 2, 1], # Simple list [3, 2, 1], # Simple list
...@@ -26,20 +26,21 @@ from tests.link.numba.test_basic import compare_numba_and_py ...@@ -26,20 +26,21 @@ from tests.link.numba.test_basic import compare_numba_and_py
["stable", UserWarning], ["stable", UserWarning],
], ],
) )
def test_Sort(x, axis, kind, exc): def test_Sort(x_test, axis, kind, exc):
x = pt.as_tensor(x_test).type("x")
if axis: if axis:
g = SortOp(kind)(pt.as_tensor_variable(x), axis) g = SortOp(kind)(x, axis)
else: else:
g = SortOp(kind)(pt.as_tensor_variable(x)) g = SortOp(kind)(x)
cm = contextlib.suppress() if not exc else pytest.warns(exc) cm = contextlib.suppress() if not exc else pytest.warns(exc)
with cm: with cm:
compare_numba_and_py([], [g], []) compare_numba_and_py([x], [g], [x_test])
@pytest.mark.parametrize( @pytest.mark.parametrize(
"x", "x_test",
[ [
[], # Empty list [], # Empty list
[3, 2, 1], # Simple list [3, 2, 1], # Simple list
...@@ -55,18 +56,19 @@ def test_Sort(x, axis, kind, exc): ...@@ -55,18 +56,19 @@ def test_Sort(x, axis, kind, exc):
["stable", UserWarning], ["stable", UserWarning],
], ],
) )
def test_ArgSort(x, axis, kind, exc): def test_ArgSort(x_test, axis, kind, exc):
if x is None: if x_test is None:
x = np.arange(5 * 5 * 5 * 5) x_test = np.arange(5 * 5 * 5 * 5)
np.random.shuffle(x) np.random.shuffle(x_test)
x = np.reshape(x, (5, 5, 5, 5)) x_test = np.reshape(x_test, (5, 5, 5, 5))
x = pt.as_tensor(x_test).type("x")
if axis: if axis:
g = ArgSortOp(kind)(pt.as_tensor_variable(x), axis) g = ArgSortOp(kind)(x, axis)
else: else:
g = ArgSortOp(kind)(pt.as_tensor_variable(x)) g = ArgSortOp(kind)(x)
cm = contextlib.suppress() if not exc else pytest.warns(exc) cm = contextlib.suppress() if not exc else pytest.warns(exc)
with cm: with cm:
compare_numba_and_py([], [g], []) compare_numba_and_py([x], [g], [x_test])
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论