def test_binarized_mnist(): parser = argparse.ArgumentParser() subparsers = parser.add_subparsers() binarized_mnist.fill_subparser(subparsers.add_parser('binarized_mnist')) args = parser.parse_args(['binarized_mnist']) sets = ['train', 'valid', 'test'] urls = ['http://www.cs.toronto.edu/~larocheh/public/datasets/' + 'binarized_mnist/binarized_mnist_{}.amat'.format(s) for s in sets] filenames = ['binarized_mnist_{}.amat'.format(s) for s in sets] assert_equal(args.filenames, filenames) assert_equal(args.urls, urls) assert args.func is default_downloader