Update theano_test.py