Don't use floating-point axes.

theano/tensor/tests/test_sort.py

@@ -165,7 +165,7 @@ def test_argsort():
 
     # Example 2
     a = tensor.dmatrix()
-    axis = tensor.scalar()
+    axis = tensor.lscalar()
     w = argsort(a, axis)
     f = theano.function([a, axis], w)
     for axis_val in 0, 1:
@@ -183,7 +183,7 @@ def test_argsort():
 
     # Example 4
     a = tensor.dmatrix()
-    axis = tensor.scalar()
+    axis = tensor.lscalar()
     l = argsort(a, axis, "mergesort")
     f = theano.function([a, axis], l)
     for axis_val in 0, 1:
@@ -193,7 +193,7 @@ def test_argsort():
 
     # Example 5
     a = tensor.dmatrix()
-    axis = tensor.scalar()
+    axis = tensor.lscalar()
     a1 = ArgSortOp("mergesort", [])
     a2 = ArgSortOp("quicksort", [])
     # All the below should give true