test_gjp_rec Namespace Reference

Functions
def	extract_values (output_str)
def	run_fortran (n, alpha, beta, method)
def	run_python (n, alpha, beta)
def	test_gjp_quad_rec (n, alpha, beta, method)

Function Documentation

extract_values()

def test_gjp_rec.extract_values

(

output_str

)

Definition at line 7 of file test_gjp_rec.py.

def extract_values(output_str):
    lines = output_str.strip().split("\n")
    roots = []
    weights = []
    for line in lines:
        _, root, _, weight = line.split()
        roots.append(float(root))
        weights.append(float(weight))
    return np.array(roots), np.array(weights)
 
 

Code

run_fortran()

def test_gjp_rec.run_fortran	(		n,
			alpha,
			beta,
			method
	)

Definition at line 18 of file test_gjp_rec.py.

def run_fortran(n, alpha, beta, method):
    result = subprocess.run(
        [
            "fpm",
            "run",
            "gjp_quad",
            "--",
            str(n),
            "{:.1f}".format(alpha),
            "{:.1f}".format(beta),
            method,
        ],
        stdout=subprocess.PIPE,
    )
    return extract_values(result.stdout.decode())
 
 

Code

run_python()

def test_gjp_rec.run_python	(		n,
			alpha,
			beta
	)

Definition at line 35 of file test_gjp_rec.py.

def run_python(n, alpha, beta):
    result = subprocess.run(
        [
            "python",
            "scripts/sympy_gauss_jac.py",
            "--npts",
            str(n),
            "--alpha",
            str(alpha),
            "--beta",
            str(beta),
        ],
        stdout=subprocess.PIPE,
    )
    return extract_values(result.stdout.decode())
 
 
@pytest.mark.parametrize(
    "n, alpha, beta, method",
    [
        (3, 1, 5, "rec"),
        (5, 2, 3, "rec"),
        pytest.param(
            10,
            0.0,
            30.0,
            "rec",
            marks=pytest.mark.xfail(reason="High beta values diverge"),
        ),
        (3, 1, 5, "gw"),
        (5, 2, 3, "gw"),
        pytest.param(
            10,
            0.0,
            30.0,
            "gw",
        ),
    ],
)

 

Code

test_gjp_quad_rec()

def test_gjp_rec.test_gjp_quad_rec	(		n,
			alpha,
			beta,
			method
	)

Definition at line 74 of file test_gjp_rec.py.

def test_gjp_quad_rec(n, alpha, beta, method):
    fortran_roots, fortran_weights = run_fortran(n, alpha, beta, method)
    python_roots, python_weights = run_python(n, alpha, beta)
 
    assert np.allclose(fortran_roots, python_roots, atol=1e-14)
    assert np.allclose(fortran_weights, python_weights, atol=1e-14)

Code