Improved test_maxout.py running speed