Skip to content

Commit

Permalink
Add tests for non cuda implementation of calc_gamma_van_der_waals().
Browse files Browse the repository at this point in the history
  • Loading branch information
@smokestacklightnin
smokestacklightnin committed Aug 17, 2023
1 parent 5a750e0 commit 95ca3c1
Showing 1 changed file with 148 additions and 0 deletions.
148 changes: 148 additions & 0 deletions stardis/opacities/tests/test_broadening.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,9 @@
calc_doppler_width,
_calc_doppler_width_cuda,
calc_doppler_width_cuda,
calc_gamma_van_der_waals,
# _calc_gamma_van_der_waals_cuda,
# calc_gamma_van_der_waals_cuda,
)

GPUs_available = cuda.is_available()
Expand Down Expand Up @@ -128,3 +131,148 @@ def test_calc_doppler_width_cuda_wrapped_sample_cuda_values(
calc_doppler_width_cuda(*map(cp.asarray, arg_list)),
calc_doppler_width_cuda_wrapped_sample_cuda_values_expected_result,
)


@pytest.mark.parametrize(
"calc_gamma_van_der_waals_sample_values_input_ion_number,calc_gamma_van_der_waals_sample_values_input_n_eff_upper,calc_gamma_van_der_waals_sample_values_input_n_eff_lower, calc_gamma_van_der_waals_sample_values_input_temperature, calc_gamma_van_der_waals_sample_values_input_h_density,calc_gamma_van_der_waals_sample_values_input_h_mass,calc_gamma_van_der_waals_sample_values_expected_result",
[
(
1, # ion_number
1.0, # n_eff_upper
0.0, # n_eff_lower
np.pi / 8 / BOLTZMANN_CONSTANT / 17 ** (1.0 / 0.3), # temperature
(3.0 * 6.46e-34) ** (-0.4), # h_density
1.0, # h_mass
1.0, # Expected output
),
# (
# np.array(2 * [1], dtype=int),
# np.array(2 * [1.0]),
# np.array(2 * [0.0]),
# np.array(2 * [np.pi / 8 / BOLTZMANN_CONSTANT / 17 ** (1./0.3)]),
# np.array(2 * [(3.0 * 6.46e-34) ** (-0.4)]),
# np.array(2 * [1.0]),
# np.array(2 * [1.0]),
# ),
],
)
def test_calc_gamma_van_der_waals_sample_values(
calc_gamma_van_der_waals_sample_values_input_ion_number,
calc_gamma_van_der_waals_sample_values_input_n_eff_upper,
calc_gamma_van_der_waals_sample_values_input_n_eff_lower,
calc_gamma_van_der_waals_sample_values_input_temperature,
calc_gamma_van_der_waals_sample_values_input_h_density,
calc_gamma_van_der_waals_sample_values_input_h_mass,
calc_gamma_van_der_waals_sample_values_expected_result,
):
print(
calc_gamma_van_der_waals(
calc_gamma_van_der_waals_sample_values_input_ion_number,
calc_gamma_van_der_waals_sample_values_input_n_eff_upper,
calc_gamma_van_der_waals_sample_values_input_n_eff_lower,
calc_gamma_van_der_waals_sample_values_input_temperature,
calc_gamma_van_der_waals_sample_values_input_h_density,
calc_gamma_van_der_waals_sample_values_input_h_mass,
)
)
assert np.allclose(
calc_gamma_van_der_waals(
calc_gamma_van_der_waals_sample_values_input_ion_number,
calc_gamma_van_der_waals_sample_values_input_n_eff_upper,
calc_gamma_van_der_waals_sample_values_input_n_eff_lower,
calc_gamma_van_der_waals_sample_values_input_temperature,
calc_gamma_van_der_waals_sample_values_input_h_density,
calc_gamma_van_der_waals_sample_values_input_h_mass,
),
calc_gamma_van_der_waals_sample_values_expected_result,
)


# @pytest.mark.skipif(
# not GPUs_available, reason="No GPU is available to test CUDA function"
# )
# @pytest.mark.parametrize(
# "calc_gamma_van_der_waals_sample_values_input_ion_number,calc_gamma_van_der_waals_sample_values_input_n_eff_upper,calc_gamma_van_der_waals_sample_values_input_n_eff_lower, calc_gamma_van_der_waals_sample_values_input_temperature, calc_gamma_van_der_waals_sample_values_input_h_density,calc_gamma_van_der_waals_sample_values_input_h_mass,calc_gamma_van_der_waals_sample_values_expected_result",
# [
# (
# np.array(2 * [1], dtype=int),
# np.array(2 * [1.0]),
# np.array(2 * [0.0]),
# np.array(2 * [np.pi / 8 / BOLTZMANN_CONSTANT / 17 ** (1.0 / 0.3)]),
# np.array(2 * [(3.0 * 6.46e-34) ** (-0.4)]),
# np.array(2 * [1.0]),
# np.array(2 * [1.0]),
# ),
# ],
# )
# def test_calc_gamma_van_der_waals_cuda_unwrapped_sample_values(
# calc_gamma_van_der_waals_sample_values_input_ion_number,
# calc_gamma_van_der_waals_sample_values_input_n_eff_upper,
# calc_gamma_van_der_waals_sample_values_input_n_eff_lower,
# calc_gamma_van_der_waals_sample_values_input_temperature,
# calc_gamma_van_der_waals_sample_values_input_h_density,
# calc_gamma_van_der_waals_sample_values_input_h_mass,
# calc_gamma_van_der_waals_sample_values_expected_result,
# ):
# arg_list = (
# calc_gamma_van_der_waals_sample_values_input_ion_number,
# calc_gamma_van_der_waals_sample_values_input_n_eff_upper,
# calc_gamma_van_der_waals_sample_values_input_n_eff_lower,
# calc_gamma_van_der_waals_sample_values_input_temperature,
# calc_gamma_van_der_waals_sample_values_input_h_density,
# calc_gamma_van_der_waals_sample_values_input_h_mass,
# )

# arg_list = tuple(map(cp.array, arg_list))
# result_values = cp.empty_like(arg_list[0])

# nthreads = 256
# length = len(calc_gamma_van_der_waals_sample_values_expected_result)
# nblocks = 1 + (length // nthreads)

# _calc_gamma_van_der_waals_cuda[nblocks, nthreads](result_values, *arg_list)

# assert np.allclose(
# cp.asnumpy(result_values),
# calc_gamma_van_der_waals_sample_values_expected_result,
# )


# @pytest.mark.skipif(
# not GPUs_available, reason="No GPU is available to test CUDA function"
# )
# @pytest.mark.parametrize(
# "calc_gamma_van_der_waals_sample_values_input_ion_number,calc_gamma_van_der_waals_sample_values_input_n_eff_upper,calc_gamma_van_der_waals_sample_values_input_n_eff_lower, calc_gamma_van_der_waals_sample_values_input_temperature, calc_gamma_van_der_waals_sample_values_input_h_density,calc_gamma_van_der_waals_sample_values_input_h_mass,calc_gamma_van_der_waals_sample_values_expected_result",
# [
# (
# np.array(2 * [1], dtype=int),
# np.array(2 * [1.0]),
# np.array(2 * [0.0]),
# np.array(2 * [np.pi / 8 / BOLTZMANN_CONSTANT / 17 ** (1.0 / 0.3)]),
# np.array(2 * [(3.0 * 6.46e-34) ** (-0.4)]),
# np.array(2 * [1.0]),
# np.array(2 * [1.0]),
# ),
# ],
# )
# def test_calc_gamma_van_der_waals_cuda_wrapped_sample_cuda_values(
# calc_gamma_van_der_waals_sample_values_input_ion_number,
# calc_gamma_van_der_waals_sample_values_input_n_eff_upper,
# calc_gamma_van_der_waals_sample_values_input_n_eff_lower,
# calc_gamma_van_der_waals_sample_values_input_temperature,
# calc_gamma_van_der_waals_sample_values_input_h_density,
# calc_gamma_van_der_waals_sample_values_input_h_mass,
# calc_gamma_van_der_waals_sample_values_expected_result,
# ):
# arg_list = (
# calc_gamma_van_der_waals_sample_values_input_ion_number,
# calc_gamma_van_der_waals_sample_values_input_n_eff_upper,
# calc_gamma_van_der_waals_sample_values_input_n_eff_lower,
# calc_gamma_van_der_waals_sample_values_input_temperature,
# calc_gamma_van_der_waals_sample_values_input_h_density,
# calc_gamma_van_der_waals_sample_values_input_h_mass,
# )
# assert np.allclose(
# calc_gamma_van_der_waals_cuda(*map(cp.asarray, arg_list)),
# calc_gamma_van_der_waals_sample_values_expected_result,
# )

0 comments on commit 95ca3c1

Please sign in to comment.