def test_visualize(self): model_dir = os.path.abspath('../data/model/20160620-173927') create_checkpoint_file(model_dir, 'model.ckpt-500000') argv = [model_dir, '--model_def', 'models.nn4' ] args = visualize.parse_arguments(argv) visualize.main(args)