#####################################################################################

#

#  Copyright (c) Microsoft Corporation. All rights reserved.

#

# This source code is subject to terms and conditions of the Apache License, Version 2.0. A 

# copy of the license can be found in the License.html file at the root of this distribution. If 

# you cannot locate the  Apache License, Version 2.0, please send an email to 

# ironpy@microsoft.com. By using this source code in any fashion, you are agreeing to be bound 

# by the terms of the Apache License, Version 2.0.

#

# You must not remove this notice, or any other, from this software.

#

#

#####################################################################################





""" This provides a more convenient harness for running this

    benchmark and collecting separate timings for each component.

"""



import sys, nt

sys.path.append([nt.environ[x] for x in nt.environ.keys() if x.lower() == "dlr_root"][0] + "\\External.LCA_RESTRICTED\\Languages\\IronPython\\27\\Lib\\Test")



def test_main(type="short"):

    import pystone

    loops = { "full": 50000, "short" : 50000, "medium" : 250000, "long" : 1000000 }[type]

    pystone.main(loops)



if __name__=="__main__":

    kind = "short"

    if len(sys.argv) > 1: kind = sys.argv[1]

    test_main(kind)