mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-27 21:55:41 -04:00
124 lines
4 KiB
Python
124 lines
4 KiB
Python
""" Full system test suite. """
|
|
|
|
from math import floor
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
import openmc
|
|
from openmc.data import JOULE_PER_EV
|
|
import openmc.deplete
|
|
|
|
from tests.regression_tests import config
|
|
from example_geometry import generate_problem
|
|
|
|
|
|
def test_full(run_in_tmpdir):
|
|
"""Full system test suite.
|
|
|
|
Runs an entire OpenMC simulation with depletion coupling and verifies
|
|
that the outputs match a reference file. Sensitive to changes in
|
|
OpenMC.
|
|
|
|
This test runs a complete OpenMC simulation and tests the outputs.
|
|
It will take a while.
|
|
|
|
"""
|
|
|
|
n_rings = 2
|
|
n_wedges = 4
|
|
|
|
# Load geometry from example
|
|
geometry, lower_left, upper_right = generate_problem(n_rings, n_wedges)
|
|
|
|
# OpenMC-specific settings
|
|
settings = openmc.Settings()
|
|
settings.particles = 100
|
|
settings.batches = 10
|
|
settings.inactive = 0
|
|
space = openmc.stats.Box(lower_left, upper_right)
|
|
settings.source = openmc.Source(space=space)
|
|
settings.seed = 1
|
|
settings.verbosity = 1
|
|
|
|
# Create operator
|
|
chain_file = Path(__file__).parents[2] / 'chain_simple.xml'
|
|
op = openmc.deplete.Operator(geometry, settings, chain_file)
|
|
op.round_number = True
|
|
|
|
# Power and timesteps
|
|
dt1 = 15.*24*60*60 # 15 days
|
|
dt2 = 1.5*30*24*60*60 # 1.5 months
|
|
N = floor(dt2/dt1)
|
|
dt = np.full(N, dt1)
|
|
power = 2.337e15*4*JOULE_PER_EV*1e6 # MeV/second cm from CASMO
|
|
|
|
# Perform simulation using the predictor algorithm
|
|
openmc.deplete.PredictorIntegrator(op, dt, power).integrate()
|
|
|
|
# Get path to test and reference results
|
|
path_test = op.output_dir / 'depletion_results.h5'
|
|
path_reference = Path(__file__).with_name('test_reference.h5')
|
|
|
|
# If updating results, do so and return
|
|
if config['update']:
|
|
shutil.copyfile(str(path_test), str(path_reference))
|
|
return
|
|
|
|
# Load the reference/test results
|
|
res_test = openmc.deplete.ResultsList.from_hdf5(path_test)
|
|
res_ref = openmc.deplete.ResultsList.from_hdf5(path_reference)
|
|
|
|
# Assert same mats
|
|
for mat in res_ref[0].mat_to_ind:
|
|
assert mat in res_test[0].mat_to_ind, \
|
|
"Material {} not in new results.".format(mat)
|
|
for nuc in res_ref[0].nuc_to_ind:
|
|
assert nuc in res_test[0].nuc_to_ind, \
|
|
"Nuclide {} not in new results.".format(nuc)
|
|
|
|
for mat in res_test[0].mat_to_ind:
|
|
assert mat in res_ref[0].mat_to_ind, \
|
|
"Material {} not in old results.".format(mat)
|
|
for nuc in res_test[0].nuc_to_ind:
|
|
assert nuc in res_ref[0].nuc_to_ind, \
|
|
"Nuclide {} not in old results.".format(nuc)
|
|
|
|
tol = 1.0e-6
|
|
for mat in res_test[0].mat_to_ind:
|
|
for nuc in res_test[0].nuc_to_ind:
|
|
_, y_test = res_test.get_atoms(mat, nuc)
|
|
_, y_old = res_ref.get_atoms(mat, nuc)
|
|
|
|
# Test each point
|
|
correct = True
|
|
for i, ref in enumerate(y_old):
|
|
if ref != y_test[i]:
|
|
if ref != 0.0:
|
|
correct = np.abs(y_test[i] - ref) / ref <= tol
|
|
else:
|
|
correct = False
|
|
|
|
assert correct, "Discrepancy in mat {} and nuc {}\n{}\n{}".format(
|
|
mat, nuc, y_old, y_test)
|
|
|
|
# Compare statepoint files with depletion results
|
|
|
|
t_test, k_test = res_test.get_eigenvalue()
|
|
t_ref, k_ref = res_ref.get_eigenvalue()
|
|
k_state = np.empty_like(k_ref)
|
|
|
|
n_tallies = np.empty(N + 1, dtype=int)
|
|
|
|
# Get statepoint files for all BOS points and EOL
|
|
for n in range(N + 1):
|
|
statepoint = openmc.StatePoint("openmc_simulation_n{}.h5".format(n))
|
|
k_n = statepoint.k_combined
|
|
k_state[n] = [k_n.nominal_value, k_n.std_dev]
|
|
n_tallies[n] = len(statepoint.tallies)
|
|
# Look for exact match pulling from statepoint and depletion_results
|
|
assert np.all(k_state == k_test)
|
|
assert np.allclose(k_test, k_ref)
|
|
|
|
# Check that no additional tallies are loaded from the files
|
|
assert np.all(n_tallies == 0)
|