# Copyright 2019 TerraPower, LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tests the geometry (loading input) file."""
import unittest
from armi.reactor import geometry
[docs]
class TestGeomType(unittest.TestCase):
[docs]
def test_fromStr(self):
# note the bonkers case and extra whitespace to exercise the canonicalization
self.assertEqual(geometry.GeomType.fromStr("HeX"), geometry.GeomType.HEX)
self.assertEqual(geometry.GeomType.fromStr("cARTESIAN"), geometry.GeomType.CARTESIAN)
self.assertEqual(geometry.GeomType.fromStr(" thetaRZ"), geometry.GeomType.RZT)
self.assertEqual(geometry.GeomType.fromStr("rZ "), geometry.GeomType.RZ)
with self.assertRaises(ValueError):
geometry.GeomType.fromStr("what even is this?")
[docs]
def test_label(self):
gt = geometry.GeomType.fromStr("hex")
self.assertEqual(gt.label, "Hexagonal")
gt = geometry.GeomType.fromStr("cartesian")
self.assertEqual(gt.label, "Cartesian")
gt = geometry.GeomType.fromStr("rz")
self.assertEqual(gt.label, "R-Z")
gt = geometry.GeomType.fromStr("thetarz")
self.assertEqual(gt.label, "R-Z-Theta")
[docs]
def test_str(self):
for geom in {geometry.HEX, geometry.CARTESIAN, geometry.RZ, geometry.RZT}:
self.assertEqual(str(geometry.GeomType.fromStr(geom)), geom)
[docs]
class TestSymmetryType(unittest.TestCase):
[docs]
def test_fromStr(self):
# note the bonkers case and extra whitespace to exercise the canonicalization
self.assertEqual(
geometry.SymmetryType.fromStr("thiRd periodic ").domain,
geometry.DomainType.THIRD_CORE,
)
st = geometry.SymmetryType.fromStr("sixteenth reflective")
self.assertEqual(st.boundary, geometry.BoundaryType.REFLECTIVE)
self.assertEqual(str(st), "sixteenth reflective")
with self.assertRaises(ValueError):
geometry.SymmetryType.fromStr("what even is this?")
[docs]
def test_fromAny(self):
st = geometry.SymmetryType.fromAny("eighth reflective through center assembly")
self.assertTrue(st.isThroughCenterAssembly)
self.assertEqual(st.domain, geometry.DomainType.EIGHTH_CORE)
self.assertEqual(st.boundary, geometry.BoundaryType.REFLECTIVE)
st = geometry.SymmetryType(geometry.DomainType.EIGHTH_CORE, geometry.BoundaryType.REFLECTIVE, True)
self.assertTrue(st.isThroughCenterAssembly)
self.assertEqual(st.domain, geometry.DomainType.EIGHTH_CORE)
self.assertEqual(st.boundary, geometry.BoundaryType.REFLECTIVE)
newST = geometry.SymmetryType.fromAny(st)
self.assertTrue(newST.isThroughCenterAssembly)
self.assertEqual(newST.domain, geometry.DomainType.EIGHTH_CORE)
self.assertEqual(newST.boundary, geometry.BoundaryType.REFLECTIVE)
[docs]
def test_baseConstructor(self):
self.assertEqual(
geometry.SymmetryType(geometry.DomainType.SIXTEENTH_CORE, geometry.BoundaryType.REFLECTIVE).domain,
geometry.DomainType.SIXTEENTH_CORE,
)
self.assertEqual(
str(geometry.SymmetryType(geometry.DomainType.FULL_CORE, geometry.BoundaryType.NO_SYMMETRY).boundary),
"",
)
[docs]
def test_label(self):
st = geometry.SymmetryType(geometry.DomainType.FULL_CORE, geometry.BoundaryType.NO_SYMMETRY)
self.assertEqual(st.domain.label, "Full")
self.assertEqual(st.boundary.label, "No Symmetry")
st = geometry.SymmetryType(geometry.DomainType.THIRD_CORE, geometry.BoundaryType.PERIODIC)
self.assertEqual(st.domain.label, "Third")
self.assertEqual(st.boundary.label, "Periodic")
st = geometry.SymmetryType(geometry.DomainType.QUARTER_CORE, geometry.BoundaryType.REFLECTIVE)
self.assertEqual(st.domain.label, "Quarter")
self.assertEqual(st.boundary.label, "Reflective")
st = geometry.SymmetryType(geometry.DomainType.EIGHTH_CORE, geometry.BoundaryType.REFLECTIVE)
self.assertEqual(st.domain.label, "Eighth")
st = geometry.SymmetryType(geometry.DomainType.SIXTEENTH_CORE, geometry.BoundaryType.REFLECTIVE)
self.assertEqual(st.domain.label, "Sixteenth")
[docs]
def test_SymmetryFactor(self):
st = geometry.SymmetryType(geometry.DomainType.FULL_CORE, geometry.BoundaryType.NO_SYMMETRY)
self.assertEqual(st.symmetryFactor(), 1.0)
st = geometry.SymmetryType(geometry.DomainType.THIRD_CORE, geometry.BoundaryType.PERIODIC)
self.assertEqual(st.symmetryFactor(), 3.0)
st = geometry.SymmetryType(geometry.DomainType.QUARTER_CORE, geometry.BoundaryType.REFLECTIVE)
self.assertEqual(st.symmetryFactor(), 4.0)
st = geometry.SymmetryType(geometry.DomainType.EIGHTH_CORE, geometry.BoundaryType.REFLECTIVE)
self.assertEqual(st.symmetryFactor(), 8.0)
st = geometry.SymmetryType(geometry.DomainType.SIXTEENTH_CORE, geometry.BoundaryType.REFLECTIVE)
self.assertEqual(st.symmetryFactor(), 16.0)
[docs]
def test_domainTypeNulls(self):
self.assertEqual(geometry.DomainType.NULL.label, "")
self.assertEqual(str(geometry.DomainType.NULL), "")
with self.assertRaises(ValueError):
geometry.DomainType.NULL.symmetryFactor()
[docs]
def test_checkValidGeomSymmetryCombo(self):
geomHex = geometry.GeomType.HEX
geomCart = geometry.GeomType.CARTESIAN
geomRZT = geometry.GeomType.RZT
geomRZ = geometry.GeomType.RZ
fullCore = geometry.SymmetryType(geometry.DomainType.FULL_CORE, geometry.BoundaryType.NO_SYMMETRY)
thirdPeriodic = geometry.SymmetryType(geometry.DomainType.THIRD_CORE, geometry.BoundaryType.PERIODIC)
quarterCartesian = geometry.SymmetryType(geometry.DomainType.QUARTER_CORE, geometry.BoundaryType.REFLECTIVE)
self.assertTrue(geometry.checkValidGeomSymmetryCombo(geomHex, thirdPeriodic))
self.assertTrue(geometry.checkValidGeomSymmetryCombo(geomHex, fullCore))
self.assertTrue(geometry.checkValidGeomSymmetryCombo(geomCart, quarterCartesian))
self.assertTrue(geometry.checkValidGeomSymmetryCombo(geomRZT, quarterCartesian))
self.assertTrue(geometry.checkValidGeomSymmetryCombo(geomRZ, fullCore))
with self.assertRaises(ValueError):
_ = geometry.SymmetryType(
geometry.DomainType.THIRD_CORE,
geometry.BoundaryType.REFLECTIVE,
False,
)
with self.assertRaises(ValueError):
geometry.checkValidGeomSymmetryCombo(geomHex, quarterCartesian)
with self.assertRaises(ValueError):
geometry.checkValidGeomSymmetryCombo(geomCart, thirdPeriodic)