Deepak Mallubhotla
071cd2503e
All checks were successful
gitea-physics/pyewjn/pipeline/head This commit looks good
58 lines
1.2 KiB
Python
58 lines
1.2 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
import pyewjn.dielectric
|
|
import pyewjn.noise.chi
|
|
from pyewjn.baskets import CalculationParams
|
|
|
|
|
|
@pytest.fixture
|
|
def chi_zz_e_lindhard():
|
|
params = CalculationParams(
|
|
omega=1e9, v_f=2e6, omega_p=3.544907701811032e15, tau=1e-14
|
|
)
|
|
eps_l = pyewjn.dielectric.get_lindhard_dielectric(params)
|
|
return pyewjn.noise.chi.get_chi_zz_e(eps_l)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_input,expected",
|
|
[
|
|
# z chi_zz_e_lindhard(z)
|
|
(1e-5, 4.0249088868003124e6),
|
|
(1e-6, 4.400474453780887e9),
|
|
(1e-7, 7.768467746685921e12),
|
|
(1e-8, 1.8541895525296864e16),
|
|
],
|
|
)
|
|
def test_chi_zz_e_lindhard(chi_zz_e_lindhard, test_input, expected):
|
|
actual = chi_zz_e_lindhard(test_input)
|
|
|
|
np.testing.assert_allclose(
|
|
actual,
|
|
expected,
|
|
rtol=1e-3,
|
|
err_msg="chi_zz_e is inaccurate for Lindhard case",
|
|
verbose=True,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_input,expected",
|
|
[
|
|
# z chi_zz_e_lindhard(z)
|
|
(1e-6, 4.400474453780887e9),
|
|
],
|
|
)
|
|
def test_chi_zz_e_lindhard_benchmark(
|
|
benchmark, chi_zz_e_lindhard, test_input, expected
|
|
):
|
|
actual = benchmark(chi_zz_e_lindhard, test_input)
|
|
np.testing.assert_allclose(
|
|
actual,
|
|
expected,
|
|
rtol=1e-3,
|
|
err_msg="chi_zz_e is inaccurate for Lindhard case",
|
|
verbose=True,
|
|
)
|