mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-21 14:35:27 -04:00
28 lines
789 B
Python
28 lines
789 B
Python
import os
|
|
|
|
import openmc
|
|
from openmc.examples import random_ray_lattice
|
|
from openmc.utility_funcs import change_directory
|
|
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("shape", ["linear", "linear_xy"])
|
|
def test_random_ray_source(shape):
|
|
with change_directory(shape):
|
|
openmc.reset_auto_ids()
|
|
model = random_ray_lattice()
|
|
model.settings.random_ray['source_shape'] = shape
|
|
model.settings.inactive = 20
|
|
model.settings.batches = 40
|
|
harness = MGXSTestHarness('statepoint.40.h5', model)
|
|
harness.main()
|