Make Pylearn2 unit-testable