@@ -0,0 +1,663 @@
+%22%22%22%0AA unit test for the summarize_model.py script%0A%22%22%22%0Aimport cPickle%0Aimport os%0A%0Afrom pylearn2.testing.skip import skip_if_no_matplotlib%0Afrom pylearn2.models.mlp import MLP, Linear%0Afrom pylearn2.scripts.summarize_model import summarize%0A%0A%0Adef test_summarize_model():%0A    %22%22%22%0A    Asks the summarize_model.py script to inspect a pickled model and%0A    check that it completes succesfully%0A    %22%22%22%0A    skip_if_no_matplotlib()%0A    with open('model.pkl', 'wb') as f:%0A        cPickle.dump(MLP(layers=%5BLinear(dim=5, layer_name='h0', irange=0.1)%5D,%0A                         nvis=10), f, protocol=cPickle.HIGHEST_PROTOCOL)%0A    summarize('model.pkl')%0A    os.remove('model.pkl')%0A

Add unit test for summarize_model.py