Add pylearn2 tests