crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-21 14:35:27 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Small changes

Browse source
This commit is contained in:
Paul Romano 2026-07-15 10:20:11 -05:00
parent 64b3309c0d
commit 55023f4d34
2 changed files with 5 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
openmc/source.py
View file

@ -958,9 +958,7 @@ class TokamakSource(SourceBase):
phi_extent : float
Toroidal angle extent in [rad] (default: 2π)
n_alpha : int
Number of poloidal angle grid points for CDF sampling (default: 101).
Values below 51 produce a warning because they may introduce noticeable
discretization bias.
Number of poloidal angle grid points for CDF sampling (default: 101)
vertical_shift : float
Vertical shift of the plasma center in [cm] (default: 0)
strength : float

13
tests/unit_tests/test_source_tokamak.py
View file

@ -103,16 +103,11 @@ def test_tokamak_source_invalid_n_alpha():
make_source(n_alpha=2)
def test_tokamak_source_low_n_alpha_warning():
with pytest.warns(UserWarning, match="below 51"):
make_source(n_alpha=50)
@pytest.mark.parametrize(("attribute", "value", "match"), [
("minor_radius", 700.0, "smaller than major_radius"),
("shafranov_shift", 150.0, "half the minor_radius"),
("emission_density", np.ones(5), "same length as r_over_a"),
("energy", [openmc.stats.Discrete([1.0], [1.0])] * 2,
("energy", [openmc.stats.delta_function(1.0)] * 2,
"Number of energy distributions"),
])
def test_tokamak_source_mutation_validation(attribute, value, match):
@ -140,7 +135,7 @@ def test_tokamak_source_radial_sampling(
shafranov_shift=0.0,
r_over_a=[0.0, 1.0],
emission_density=emission_density,
energy=openmc.stats.Discrete([14.07e6], [1.0]),
energy=openmc.stats.delta_function(14.07e6),
)
sphere = openmc.Sphere(r=2000.0, boundary_type='vacuum')
@ -170,7 +165,7 @@ def test_tokamak_source_poloidal_sampling(run_in_tmpdir):
shafranov_shift=0.0,
emission_density=np.ones(10),
n_alpha=3,
energy=openmc.stats.Discrete([14.07e6], [1.0]),
energy=openmc.stats.delta_function(14.07e6),
)
sphere = openmc.Sphere(r=2000.0, boundary_type='vacuum')
@ -215,7 +210,7 @@ def test_tokamak_source_sampling(run_in_tmpdir):
triangularity=delta, shafranov_shift=shafranov, vertical_shift=zshift,
r_over_a=r_over_a, emission_density=1.0 - r_over_a**2,
phi_start=phi_start, phi_extent=phi_extent, n_alpha=201,
energy=openmc.stats.Discrete([14.07e6], [1.0]))
energy=openmc.stats.delta_function(14.07e6))
sphere = openmc.Sphere(r=2000.0, boundary_type='vacuum')
cell = openmc.Cell(region=-sphere)