import sys
import timeit

def load_file(unpickler):
    """Load the pickle file using module unpickler."""
    pickle = __import__(unpickler)
    with open("test.pickle", 'rb') as handle:
        d = pickle.load(handle)

if __name__ == "__main__":

    # usage: python2.x load_file.py unpickler gc_state

    unpickler = sys.argv[1] # name of pickle module to use (pickle or 
                            # cPickle)

    gc_state = sys.argv[2] # garbage collection on/off (gc_on or gc_off)

    stmt = "load_file('%s')" % unpickler
    setup = "from __main__ import load_file"

    if gc_state == 'gc_on':
        setup = "gc.enable(); " + setup

    timer = timeit.Timer(stmt, setup=setup)
    time = timer.timeit(number=1)
    print float("%.3g" % time)