OpenMC/tests/regression_tests/random_ray_volume_estimator/test.py
John Tramm 5bc04b5d78
Alternative Random Ray Volume Estimators (#3060)
Co-authored-by: Olek <45364492+yardasol@users.noreply.github.com>
2024-08-23 11:56:26 -05:00

30 lines
904 B
Python

import os
import openmc
from openmc.utility_funcs import change_directory
from openmc.examples import random_ray_three_region_cube
import pytest
from tests.testing_harness import TolerantPyAPITestHarness
class MGXSTestHarness(TolerantPyAPITestHarness):
def _cleanup(self):
super()._cleanup()
f = 'mgxs.h5'
if os.path.exists(f):
os.remove(f)
@pytest.mark.parametrize("estimator", ["hybrid",
"simulation_averaged",
"naive"
])
def test_random_ray_volume_estimator(estimator):
with change_directory(estimator):
openmc.reset_auto_ids()
model = random_ray_three_region_cube()
model.settings.random_ray['volume_estimator'] = estimator
harness = MGXSTestHarness('statepoint.10.h5', model)
harness.main()