new_contents: """
  A unit test for the train.py script
  """
  import os
  
  import pylearn2
  from pylearn2.scripts.train import train
  
  
  def test_train_cmd():
      """
      Calls the train.py script with a short YAML file
      to see if it trains without error
      """
      train(os.path.join(pylearn2.__path__[0],
                         "scripts/autoencoder_example/dae.yaml"))
  
subject: Add unit test for train.py