test.py

from __future__ import print_function

import argparse
import os
import sys
import subprocess
import time


def run_test(path):
    os.chdir(path)
    
    print('cwd', os.getcwd())
    start = time.time()
    subprocess.call('wget -nv http://download.tensorflow.org/example_images/flower_photos.tgz', shell=True) 
    end = time.time()
    print("download", end - start)
    
    start = time.time()
    print('calling tar')
    subprocess.call('tar xzf flower_photos.tgz', shell=True)
    end = time.time()
    print("extract", end - start)
    
if __name__ == '__main__':
    subprocess.call('apt-get install wget', shell=True)
    cwd = os.getcwd()
    
    parser = argparse.ArgumentParser()
    parser.add_argument(
      '--data',
      type=str)
    
    FLAGS, unparsed = parser.parse_known_args()
  
    run_test('/tmp')
    run_test(cwd)
    run_test(FLAGS.data)