Skip to content

Commit adf3505

Browse files
committed
Reintroduce pad tests
1 parent 5fcddb6 commit adf3505

File tree

1 file changed

+1
-13
lines changed

1 file changed

+1
-13
lines changed

tests/link/numba/test_pad.py

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -8,24 +8,14 @@
88

99

1010
floatX = config.floatX
11-
RTOL = ATOL = 1e-6 if floatX.endswith("64") else 1e-3
1211

1312

14-
@pytest.mark.skip(
15-
"Numba compilation is too slow and this is just a test for dispatch of OpFromGraph"
16-
)
1713
@pytest.mark.parametrize(
1814
"mode, kwargs",
1915
[
2016
("constant", {"constant_values": 0}),
2117
("constant", {"constant_values": (1, 2)}),
22-
pytest.param(
23-
"edge",
24-
{},
25-
marks=pytest.mark.skip(
26-
"This is causing a segfault in NUMBA mode, but I have no idea why"
27-
),
28-
),
18+
("edge", {}),
2919
("linear_ramp", {"end_values": 0}),
3020
("linear_ramp", {"end_values": (1, 2)}),
3121
("reflect", {"reflect_type": "even"}),
@@ -65,6 +55,4 @@ def test_numba_pad(mode: PadMode, kwargs):
6555
[x_pt],
6656
[res],
6757
[x],
68-
assert_fn=lambda x, y: np.testing.assert_allclose(x, y, rtol=RTOL, atol=ATOL),
69-
py_mode="FAST_RUN",
7058
)

0 commit comments

Comments
 (0)