Add test for slicing symbolic vars

MudeLin · Jul 24, 2015 · 5f6e897 · 5f6e897
1 parent 27ecbc0
commit 5f6e897
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/theano/tensor/tests/test_basic.py b/theano/tensor/tests/test_basic.py
@@ -7729,6 +7729,13 @@ def test_allocempty():
     assert out.shape == (2, 3)
     assert out.dtype == 'float32'
 
+
+def test_symbolic_slice():
+    x = theano.tensor.tensor4('x')
+    a, b = x.shape[:2]
+    output = a.eval({x: numpy.zeros((5, 4, 3, 2), dtype=theano.config.floatX)})
+    assert output == numpy.array(5)
+
 """
 
 if __name__ == '__main__':