small test fix

After that I merge your PR in Theano. I'll add some documentation in another PR directly to Theano.