Add numba-scipy requirement and test a SciPy-based Elemwise in Numba

requirements.txt

@@ -14,6 +14,7 @@ versioneer
 jax; python_version > '3.6'
 jaxlib; python_version > '3.6'
 numba
+numba-scipy>=0.3.0
 diff-cover
 pre-commit
 isort

tests/link/test_numba.py

@@ -93,11 +93,16 @@ def compare_numba_and_py(
 @pytest.mark.parametrize(
     "inputs, input_vals, output_fn",
     [
+        (
+            [aet.vector()],
+            [np.random.randn(100).astype(config.floatX)],
+            lambda x: aet.nnet.sigmoid(x),
+        ),
         (
             [aet.vector() for i in range(4)],
             [np.random.randn(100).astype(config.floatX) for i in range(4)],
             lambda x, y, x1, y1: (x + y) * (x1 + y1) * y,
-        )
+        ),
     ],
 )
 def test_Elemwise(inputs, input_vals, output_fn):