Laboratoire de l'Informatique et du Parallélisme » Pobyso

"""

Test module for pobyso.py functions.

AUTHORS:

- S.T. (2014-02-26): initial version

TODO:

- almost everything!

"""

print "testPobyso loading..."

#

import inspect

import resource

import timeit

def memory_usage_resource():

    rusage_denom = RR("1024")

    if sys.platform == 'darwin':

        # ... it seems that in OSX the output is different units ...

        rusage_denom = rusage_denom^2

    mem = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / rusage_denom

    return mem

# End memory_usage_resource()

def test_pobyso_wrapper(func, *args, **kwargs):

    def wrapped():

        return func(*args, **kwargs)

    return wrapped

#

def test_pobyso_absolute_so_so(repeat=10000, number=100):

    functionName =  inspect.stack()[0][3]

    print "Running", inspect.stack()[0][3], "..."

    #

    def test():

        absoluteSo = pobyso_absolute_so_so()

        sollya_lib_clear_obj(absoluteSo)

    #

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_absolute_so_so

def test_pobyso_autoprint_so_so(iterationsNum=1000):

    absoluteSo = pobyso_absolute_so_so()

    for index in xrange(0,iterationsNum):

        pobyso_autoprint(absoluteSo)

    sollya_lib_clear_obj(absoluteSo)

# End test_pobyso_autoprint

def test_pobyso_bounds_to_range_sa_so(repeat=1000, number=10):

    functionName =  inspect.stack()[0][3]

    print "Running", inspect.stack()[0][3], "..."

    lowerBoundSa = RR(1)

    upperBoundSa = RR(2)

    precSo = pobyso_get_prec_so()

    #pobyso_autoprint(precSo)

    #

    def test(lowerBoundSa, upperBoundSa):

        rangeSo = pobyso_bounds_to_range_sa_so(lowerBoundSa, upperBoundSa)

        sollya_lib_clear_obj(rangeSo)

    #

    wrapped = test_pobyso_wrapper(test,lowerBoundSa, upperBoundSa)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    #

    precSo = pobyso_get_prec_so()

    #pobyso_autoprint(precSo)

    sollya_lib_clear_obj(precSo)

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_bounds_to_range_sa_so

def test_pobyso_clear_full_list_elements_sa_so(repeat=1000, number=10):

    functionName =  inspect.stack()[0][3]

    print "Running", inspect.stack()[0][3], "..."

    intervalSa = RIF(1,2)

    functionSo = pobyso_parse_string("exp(x)")

    degreeSo   = pobyso_constant_from_int_sa_so(4)

    #RealIntervalField

    pointSo    = pobyso_constant_sa_so(RR("1.5"))

    intervalSo = pobyso_interval_to_range_sa_so(intervalSa)

    #

    def test(functionSo, degreeSo, pointSo, intervalSo):

        """

        We use Taylor models to generate the lists.

        """

        taylorformSo = pobyso_taylorform_so_so(functionSo,

                                               degreeSo,

                                               pointSo,

                                               intervalSo)

        taylorformSaSo = pobyso_get_list_elements_so_so(taylorformSo)

        pobyso_clear_full_list_elements_sa_so(taylorformSaSo)

        sollya_lib_clear_obj(taylorformSo)

    #

    wrapped = test_pobyso_wrapper(test,

                                  functionSo,

                                  degreeSo,

                                  pointSo,

                                  intervalSo)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    #

    sollya_lib_clear_obj(functionSo)

    sollya_lib_clear_obj(degreeSo)

    sollya_lib_clear_obj(pointSo)

    sollya_lib_clear_obj(intervalSo)

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_clear_full_list_elements_sa_so

def test_pobyso_float_poly_sa_so(repeat=100, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName,"..."

    polynomialsList = []

    i = var('i')

    RRR         = RealField(12)

    P_RRR       = RRR[i]

    polynomialsList.append(P_RRR(1/7 + 2/7*i + 3/7*i^2))

    RRR         = RealField(53)

    P_RRR       = RRR[i]

    polynomialsList.append(P_RRR(1/7 + 2/7*i + 3/7*i^2))

    RRR         = RealField(1024)

    P_RRR       = RRR[i]

    polynomialsList.append(P_RRR(1/7 + 2/7*i + 3/7*i^2))

    def test(polynomialsList):

        for polynomial in polynomialsList:

            polySo = pobyso_float_poly_sa_so(polynomial)

            sollya_lib_clear_obj(polySo)

    wrapped = test_pobyso_wrapper(test,

                                  polynomialsList)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_float_poly_sa_so

def test_pobyso_error_so(repeat=1000, number=10):

    functionName =  inspect.stack()[0][3]

    print "Running", inspect.stack()[0][3], "..."

    #

    def test():

        errorSo = pobyso_error_so()

        sollya_lib_clear_obj(errorSo)

    #

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_error_so

def test_pobyso_get_list_elements_so_so(repeat = 1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    def test():

        listSo = pobyso_parse_string_sa_so("[|0,1,2,3,4,5,6|];")

        (listSaSo, elementsNumSa, isEndEllipticSa) = \

            pobyso_get_list_elements(listSo)

        for elementSo in listSaSo:

            sollya_lib_clear_obj(elementSo)

        # Notice that we use sollya_lib_clear_obj and not

        # sollya_lib_clear_object_list.

        sollya_lib_clear_obj(listSo, None)

    #

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_get_list_elements

def test_pobyso_get_prec_so_sa(repeat=1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    def test():

        precSa = pobyso_get_prec_so_sa()

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_get_prec_so_sa

def test_pobyso_get_prec_of_constant_so_sa(repeat=1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    constSo = pobyso_constant_1_sa_so()

    def test():

        precSa = pobyso_get_prec_of_constant_so_sa(constSo)

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    sollya_lib_clear_obj(constSo,None)

    return timing

# End test_pobyso_get_prec_of_constant_so_sa

def test_pobyso_guess_degree_sa_sa(repeat=10, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    funcExpSa(x) = exp(x)

    RRIF = RealIntervalField(113)

    intervalSa = RRIF(3/8 - 2^-68, 3/8 + 2^-68)

    print "Interval:", intervalSa.str(style='brackets')

    errorSa = RR(2^-227)

    weightSa(x) = 1

    degreeBoundSa = 150

    # Function definition

    def test():

        pobyso_guess_degree_sa_sa(funcExpSa,

                                  intervalSa,

                                  errorSa,

                                  weightSa,

                                  degreeBoundSa)

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_guess_degree_sa_sa

def test_pobyso_guess_degree_so_sa(repeat=1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    funcExpSo = pobyso_parse_string_sa_so("exp(x)")

    RRIF = RealIntervalField(113)

    intervalSa = RRIF(3/8 - 2^-68, 3/8 + 2^-68)

    print "Interval:", intervalSa.str(style='brackets')

    intervalSo = pobyso_interval_to_range_sa_so(intervalSa)

    errorSo = pobyso_constant_sa_so(RR(2^-227))

    weightSo=pobyso_parse_string_sa_so("1")

    degreeBoundSo = pobyso_constant_from_int_sa_so(150)

    # Function definition

    def test():

        pobyso_guess_degree_so_sa(funcExpSo,

                                  intervalSo,

                                  errorSo,

                                  weightSo,

                                  degreeBoundSo)

    """

        pobyso_guess_degree_so_sa(funcExpSo,

                                  intervalSo,

                                  errorSo,

                                  None,

                                  None)

    """

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    sollya_lib_clear_obj(funcExpSo)

    sollya_lib_clear_obj(intervalSo)

    sollya_lib_clear_obj(errorSo)

    sollya_lib_clear_obj(weightSo)

    sollya_lib_clear_obj(degreeBoundSo)

    return timing

# End test_pobyso_guess_degree_so_sa

def test_pobyso_is_error_so_sa(repeat=1000, number=10):

    functionName =  inspect.stack()[0][3]

    print "Running", inspect.stack()[0][3], "..."

    #

    def test():

        errorSo = pobyso_error_so()

        if not pobyso_is_error_so_sa(errorSo):

            print "Error!"

        sollya_lib_clear_obj(errorSo)

    #

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_error_so

def test_pobyso_lib_init(repeat=1000, number=10):

    """

    Must be called exactly once. Leaks lots of memory otherwise.

    """

    functionName = inspect.stack()[0][3]

    print "Running", functionName,"..."

    def test():

        pobyso_lib_init()

        pobyso_lib_close()

    wrapped = test_pobyso_wrapper(test)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_lib_init

def test_pobyso_parse_string_sa_so(repeat=1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName,"..."

    stringToParse = "exp(x)^(2)/sin(x)*cos(x)"

    def test(stringToParseSa):

        expressionSo = pobyso_parse_string_sa_so(stringToParseSa)

        sollya_lib_clear_obj(expressionSo)

    wrapped = test_pobyso_wrapper(test,

                                  stringToParse)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_parse_string_sa_so

def test_pobyso_rat_poly_sa_so(repeat=100, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName,"..."

    precisionsList = [12, 53, 163, 1024]

    i = var('i')

    P_QQ        = QQ[i]

    pr          = P_QQ(1/7 + 2/7*i + 3/7*i^2)

    def test(polynomial, precisionsList):

        for prec in precisionsList:

            polySo = pobyso_rat_poly_sa_so(polynomial, prec)

            sollya_lib_clear_obj(polySo)

    wrapped = test_pobyso_wrapper(test,

                                  pr,

                                  precisionsList)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_rat_poly_sa_so

def test_pobyso_taylor_expansion_no_change_var_so_so(repeat=1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    functionSo  = pobyso_parse_string('exp(x)')

    degreeSo    = pobyso_constant_from_int_sa_so(20)

    rangeSo     = pobyso_bounds_to_range_sa_so(RR(1),RR(2))

    errorTypeSo = pobyso_absolute_so_so()

    def test(functionSo, degreeSo, rangeSo, errorTypeSo):

        teSo = pobyso_taylor_expansion_no_change_var_so_so(functionSo,

                                                           degreeSo,

                                                           rangeSo,

                                                           errorTypeSo)

        sollya_lib_clear_obj(teSo[0])

        sollya_lib_clear_obj(teSo[1])

        sollya_lib_clear_obj(teSo[2])

    # End test

    wrapped = test_pobyso_wrapper(test,

                                  functionSo,

                                  degreeSo,

                                  rangeSo,

                                  errorTypeSo)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    sollya_lib_clear_obj(functionSo)

    sollya_lib_clear_obj(degreeSo)

    sollya_lib_clear_obj(rangeSo)

    sollya_lib_clear_obj(errorTypeSo)

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_taylor_expansion_no_change_var_so_so

def test_pobyso_taylor_expansion_with_change_var_so_so(repeat=1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    functionSo  = pobyso_parse_string('exp(x)')

    degreeSo    = pobyso_constant_from_int_sa_so(20)

    rangeSo     = pobyso_bounds_to_range_sa_so(RR(1),RR(2))

    errorTypeSo = pobyso_absolute_so_so()

    def test(functionSo, degreeSo, rangeSo, errorTypeSo):

        teSo = pobyso_taylor_expansion_with_change_var_so_so(functionSo,

                                                             degreeSo,

                                                             rangeSo,

                                                             errorTypeSo)

        sollya_lib_clear_obj(teSo[0])

        sollya_lib_clear_obj(teSo[1])

        sollya_lib_clear_obj(teSo[2])

    # End test

    wrapped = test_pobyso_wrapper(test,

                                  functionSo,

                                  degreeSo,

                                  rangeSo,

                                  errorTypeSo)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    sollya_lib_clear_obj(functionSo)

    sollya_lib_clear_obj(degreeSo)

    sollya_lib_clear_obj(rangeSo)

    sollya_lib_clear_obj(errorTypeSo)

    print "\t...", functionName, "done."

    return timing

# End test_pobyso_taylor_expansion_with_change_var_so_so

def test_pobyso_taylorform_so_so_2(repeat=1000, number=10):

    functionName = inspect.stack()[0][3]

    print "Running", functionName, "..."

    functionSo  = pobyso_parse_string('exp(x)')

    degreeSo    = pobyso_constant_from_int_sa_so(20)

    rangeSo     = pobyso_bounds_to_range_sa_so(RR(1),RR(2))

    intervalCenterSo = sollya_lib_mid(rangeSo, None)

    errorTypeSo = pobyso_absolute_so_so()

    def test(functionSo, degreeSo, intervalCenterSo, errorTypeSo):

        taylorformSo = pobyso_taylorform_so_so(functionSo,

                                               degreeSo,

                                               intervalCenterSo,

                                               errorTypeSo)

        sollya_lib_clear_obj(taylorformSo)

    wrapped = test_pobyso_wrapper(test,

                                  functionSo,

                                  degreeSo,

                                  intervalCenterSo,

                                  errorTypeSo)

    timing = min(timeit.repeat(wrapped, repeat=repeat, number=number))

    sollya_lib_clear_obj(functionSo, None)

    sollya_lib_clear_obj(degreeSo, None)

    sollya_lib_clear_obj(rangeSo, None)

    sollya_lib_clear_obj(intervalCenterSo, None)

    sollya_lib_clear_obj(errorTypeSo, None)

    return timing

# End   test_pobyso_taylor_so_so_2

def test_pobyso_dummy_test(iterationsNum=1, executionsNum=1):

    aString = "2+3"

    def test():

        return 2+3

    wrapped = test_pobyso_wrapper(test)

    timing = timeit.timeit(wrapped, number=executionsNum)

    return timing

print "\t...testPobyso loaded"