Example #1
0
def sols_to_co(sols):
    sol, order = sols
    src = ""
    q = sol.keys()
    for var in order:
        src = "%s = %s\n" % (str(var), numpy_print(sol[var])) \
            + src
        q.remove(var)
    reject = []
    len_reject = 0
    while len(q) != 0 or len(reject) != 0:
        if len(q) == 0:
            if len_reject == len(reject):
                break
            len_reject = len(reject)
            q = reject
            reject = []
        var = q.pop(0)
        rejected = False
        for k in q:
            if var in sol[k]:
                reject.append(var)
                rejected = True
                break
        if rejected:
            continue
        src = "%s = %s\n" % (str(var), numpy_print(sol[var])) + src
    if len(reject) != 0:
        raise ValueError("Sols has circular dependency,\n%s" % \
                         str((reject, sol)))
    return compile(src, "<sols_to_co>", "exec")
def sols_to_co(sols):
    sol, order = sols
    src = ""
    q = sol.keys()
    for var in order:
        src = "%s = %s\n" % (str(var), numpy_print(sol[var])) \
            + src
        q.remove(var)
    reject = []
    len_reject = 0
    while len(q) != 0 or len(reject) != 0:
        if len(q) == 0:
            if len_reject == len(reject):
                break
            len_reject = len(reject)
            q = reject
            reject = []
        var = q.pop(0)
        rejected = False
        for k in q:
            if var in sol[k]:
                reject.append(var)
                rejected = True
                break
        if rejected:
            continue
        src = "%s = %s\n" % (str(var), numpy_print(sol[var])) + src
    if len(reject) != 0:
        raise ValueError("Sols has circular dependency,\n%s" % \
                         str((reject, sol)))
    return compile(src, "<sols_to_co>", "exec")
def test_numpy_print():
    assert(numpy_print(-mu_12 * p_1 + r_2) == "-dot(mu_12, p_1) + r_2")
    assert(numpy_print(delta_1 * A * p_1 + r_1) == "dot(delta_1, dot(A, p_1)) + r_1")
    assert(numpy_print(delta_1 * p_1 + x_1) == "dot(delta_1, p_1) + x_1")
    assert(numpy_print((T(r_1) * r_1) / (T(r_1) * A * p_1)) == \
                       "dot(dot((r_1).transpose(), r_1), 1.0 / (dot(dot((p_1).transpose(), A), r_1)))")
    assert(numpy_print((T(p_1) * A * r_2) / (T(p_1) * A * p_1)) == \
                       "dot(dot(dot((p_1).transpose(), A), r_2), 1.0 / (dot(dot((p_1).transpose(), A), p_1)))")
Example #4
0
def test_numpy_print():
    assert (numpy_print(-mu_12 * p_1 + r_2) == "-dot(mu_12, p_1) + r_2")
    assert (numpy_print(delta_1 * A * p_1 +
                        r_1) == "dot(delta_1, dot(A, p_1)) + r_1")
    assert (numpy_print(delta_1 * p_1 + x_1) == "dot(delta_1, p_1) + x_1")
    assert(numpy_print((T(r_1) * r_1) / (T(r_1) * A * p_1)) == \
                       "dot(dot((r_1).transpose(), r_1), 1.0 / (dot(dot((p_1).transpose(), A), r_1)))")
    assert(numpy_print((T(p_1) * A * r_2) / (T(p_1) * A * p_1)) == \
                       "dot(dot(dot((p_1).transpose(), A), r_2), 1.0 / (dot(dot((p_1).transpose(), A), p_1)))")