crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 06:05:58 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Parallelize sampling external sources and threadsafe rejection counters (#3830)

Browse source 
Co-authored-by: Paul Romano <paul.k.romano@gmail.com>
This commit is contained in:
Ethan Peterson 2026-03-04 15:36:43 -05:00 • committed by GitHub
parent 0ab46dfa35
commit 2bd06660c5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 114 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/unit_tests/test_lib.py
View file

@ -1114,6 +1114,14 @@ def test_sample_external_source(run_in_tmpdir, mpi_intracomm):
assert p1.time == p2.time
assert p1.wgt == p2.wgt
# as_array should return a numpy structured array with matching values
arr = openmc.lib.sample_external_source(10, prn_seed=3, as_array=True)
assert isinstance(arr, np.ndarray)
assert len(arr) == 10
for p, row in zip(particles, arr):
assert p.r == pytest.approx(row['r'])
assert p.E == pytest.approx(row['E'])
openmc.lib.finalize()
# Make sure sampling works in volume calculation mode