pyewjn/tests/dielectric/test_lindhard_dielectric_transverse.py
Deepak Mallubhotla 6ecb74f1b9
All checks were successful
gitea-physics/pyewjn/pipeline/head This commit looks good
feat: adds r_s and chi_zz_b functions
2022-06-08 09:37:39 -05:00

28 lines
786 B
Python

import pyewjn.dielectric
import numpy as np
import pytest
from pyewjn.baskets import CalculationParams
def get_common_lindhard_dielectric():
params = CalculationParams(omega=1e9, omega_p=3.5e15, tau=1e-14, v_f=2e6)
return pyewjn.dielectric.get_lindhard_dielectric_transverse(params)
@pytest.mark.skip(reason="Not actually correct values")
@pytest.mark.parametrize(
"test_input,expected",
[
(10, -1222.185185062794 + 1.2249999998777178e8j),
(1000, 16924.14814718176 + 1.2250000020552777e8j),
(1e8, 83.687499999706 + 0.00022417398943752126j),
],
)
def test_lindhard_dielectric_transverse(test_input, expected):
eps_to_test = get_common_lindhard_dielectric()
np.testing.assert_almost_equal(
eps_to_test(test_input), expected, decimal=6, err_msg="b function is off"
)