tests: added dynamic testing

Author: BitterB0NG0

.gitignore

@@ -116,7 +116,7 @@ target/
 
 # IPython
 profile_default/
-ipython_config.py
+ipython_utils.py
 
 # pyenv
 #   For a library or package, you might want to ignore these files since the code is

examples/data/input_files/F1F8.i renamed to examples/data/inp/F1F8.i

@@ -1,38 +1,79 @@
 import pymcnp
 
 
-def get_inp(RADIUS_AIR, RADIUS_SHIELD, RADIUS_LEAD):
-    BOX_AIR = pymcnp.inp.SurfaceOption__Rpp.from_mcnp(
-        f'rpp {-RADIUS_AIR} {RADIUS_AIR} {-RADIUS_AIR} {RADIUS_AIR} {-RADIUS_AIR} {RADIUS_AIR}'
+def get_inp(radius_air, radius_shield, radius_lead):
+    inp = pymcnp.InpBuilder(title='Box Example')
+
+    geometry_air = pymcnp.GeometryBuilder('11')
+    geometry_shield = pymcnp.GeometryBuilder('12')
+    geometry_lead = pymcnp.GeometryBuilder('13')
+    geometry_world = pymcnp.GeometryBuilder('14')
+
+    cell_air = pymcnp.CellBuilder(number=1, material=21, density=0.5, geometry=geometry_air)
+
+    cell_shield = pymcnp.CellBuilder(
+        number=2, material=22, density=0.5, geometry=geometry_shield & geometry_air
+    )
+
+    cell_lead = pymcnp.CellBuilder(
+        number=3, material=23, density=0.5, geometry=geometry_lead & geometry_shield
     )
-    BOX_SHIELD = pymcnp.inp.SurfaceOption__Rpp.from_mcnp(
-        f'rpp {-RADIUS_SHIELD} {RADIUS_SHIELD} {-RADIUS_SHIELD} {RADIUS_SHIELD} {-RADIUS_SHIELD} {RADIUS_SHIELD}'
+
+    cell_world = pymcnp.CellBuilder(number=4, material=0, geometry=geometry_world)
+
+    inp.append(cell_air)
+    inp.append(cell_shield)
+    inp.append(cell_lead)
+    inp.append(cell_world)
+
+    surface_air = pymcnp.SurfaceBuilder(
+        number=11,
+        mnemonic='rpp',
+        parameter=f'{-radius_air} {radius_air} {-radius_air} {radius_air} {-radius_air} {radius_air}',
     )
-    BOX_LEAD = pymcnp.inp.SurfaceOption__Rpp.from_mcnp(
-        f'rpp {-RADIUS_LEAD} {RADIUS_LEAD} {-RADIUS_LEAD} {RADIUS_LEAD} {-RADIUS_LEAD} {RADIUS_LEAD}'
+    surface_shield = pymcnp.SurfaceBuilder(
+        number=12,
+        mnemonic='rpp',
+        parameter=f'{-radius_shield} {radius_shield} {-radius_shield} {radius_shield} {-radius_shield} {radius_shield}',
+    )
+    surface_lead = pymcnp.SurfaceBuilder(
+        number=13,
+        mnemonic='rpp',
+        parameter=f'{-radius_lead} {radius_lead} {-radius_lead} {radius_lead} {-radius_lead} {radius_lead}',
+    )
+    surface_world = pymcnp.SurfaceBuilder(
+        number=14,
+        mnemonic='so',
+        parameter=f'{radius_air + radius_shield + radius_lead + 1}',
     )
 
-    MATERIAL_AIR = pymcnp.inp.DataOption__M.from_formula(1, {'N2': 0.8, 'O2': 0.2})
-    MATERIAL_SHIELD = pymcnp.inp.DataOption__M.from_formula(2, {'TiO2': 0.5, 'PbO': 0.5})
-    MATERIAL_LEAD = pymcnp.inp.DataOption__M.from_formula(3, {'Pb': 1})
+    inp.append(surface_air)
+    inp.append(surface_shield)
+    inp.append(surface_lead)
+    inp.append(surface_world)
 
-    return f"""
-Box Example
-10 1 0.5 20
-11 2 0.5 21:20
-12 3 0.5 22:21
-13 0     23
+    material_air = pymcnp.DataBuilder.unbuild(
+        pymcnp.inp.Data(pymcnp.inp.data.M.from_formula(21, {'N2': 0.8, 'O2': 0.2}))
+    )
+    material_shield = pymcnp.DataBuilder.unbuild(
+        pymcnp.inp.Data(pymcnp.inp.data.M.from_formula(22, {'TiO2': 0.5, 'PbO': 0.5}))
+    )
+    material_lead = pymcnp.DataBuilder.unbuild(
+        pymcnp.inp.Data(pymcnp.inp.data.M.from_formula(23, {'Pb': 1}))
+    )
+
+    sdef = pymcnp.DataBuilder(
+        mnemonic='sdef',
+        parameter='x=0 y=0 z=0 erg=2.2',
+    )
 
-20 {BOX_AIR}
-21 {BOX_SHIELD}
-22 {BOX_LEAD}
-23 SO {RADIUS_AIR + RADIUS_SHIELD + RADIUS_LEAD + 1}
+    inp.append(material_air)
+    inp.append(material_shield)
+    inp.append(material_lead)
+    inp.append(sdef)
 
-{MATERIAL_AIR}
-{MATERIAL_SHIELD}
-{MATERIAL_LEAD}
-SDEF X=0 Y=0 Z=0 ERG=2.2
-"""[1:-1]
+    return inp.build()
 
 
-print(get_inp(60, 15, 1))
+inp = get_inp(60, 15, 1)
+print(inp)

examples/data/input_files/F4-fmesh-time-energy.i renamed to examples/data/inp/F4-fmesh-time-energy.i

@@ -0,0 +1,8 @@
+import pathlib
+
+import pymcnp
+
+
+inp = pymcnp.Inp.from_mcnp_file(pathlib.Path(__file__).parent / 'data/input_files/png.i')
+vista = inp.to_pyvista()
+vista.plot()

examples/data/input_files/F4-fmesh-time.i renamed to examples/data/inp/F4-fmesh-time.i

@@ -50,7 +50,7 @@ def _REGEX(element):
                 o += f'(( {{{attribute.type[12:-1]}._REGEX.pattern}})+)'
         else:
             if 'Option_' in attribute.type:
-                o += f'( {{{SNAKE(element.name)}.{CAMEL(element.name)}Option_._REGEX.pattern}})'
+                o += f'( ({{{SNAKE(element.name)}.{CAMEL(element.name)}Option_._REGEX.pattern}}))'
             else:
                 o += f'( {{{attribute.type}._REGEX.pattern}})'
 

examples/data/input_files/F4-fmesh.i renamed to examples/data/inp/F4-fmesh.i

@@ -25,7 +25,7 @@ class Test_{CAMEL(parent_name)}{CAMEL(element.name)}:
     Tests ``{CAMEL(parent_name)}{CAMEL(element.name)}``.
     """
 
-    class Test_FromMcnp(_Test_FromMcnp):
+    class Test_FromMcnp(_utils._Test_FromMcnp):
         """
         Tests ``{CAMEL(parent_name)}{CAMEL(element.name)}.from_mcnp``.
         """
@@ -58,41 +58,13 @@ def get_tests(element, mod, parent_name):
 for card in inp_data.cards.options:
     tests = get_tests(card, '', '')
 
-    res = f'''
+    res = f"""
 import pymcnp
-
-import pytest
-
-
-class _Test_FromMcnp:
-    """
-    Tests ``McnpElement_.from_mcnp``.
-    """
-
-    element: pymcnp.utils._object.McnpElement_
-    EXAMPLE_VALID: list[str]
-    EXAMPLE_INVALID: list[str]
-
-    def test_valid(self):
-        """
-        Tests ``EXAMPLES_VALID``.
-        """
-
-        for example in self.EXAMPLES_VALID:
-            self.element.from_mcnp(example)
-
-    def test_invalid(self):
-        """
-        Tests ``EXAMPLES_INVALID``.
-        """
-
-        for example in self.EXAMPLES_INVALID:
-            with pytest.raises(pymcnp.utils.errors.InpError):
-                self.element.from_mcnp(example)
+import _utils
 
 
 {'\n\n'.join(tests)}
-'''[1:]
+"""[1:]
 
     path_file = pathlib.Path(__file__).parent.parent / 'tests' / f'test_{card.name}.py'
     with path_file.open('w') as file:

examples/data/input_files/png.i renamed to examples/data/inp/png.i

@@ -1,3 +1,4 @@
+import math
 import dataclasses
 
 from . import inp
@@ -17,6 +18,20 @@ class GeometryBuilder:
 
     infix: str
 
+    @staticmethod
+    def unbuild(geometry: types.Geometry):
+        """
+        Generates ``GeometryBuilder`` from ``Geometry``.
+
+        Parameter:
+            geoemtry: ``Geometry`` to unbuild.
+
+        Returns:
+            ``GeometryBuilder`` for ``Geometry``.
+        """
+
+        return GeometryBuilder(infix=geometry.infix)
+
     def __and__(a, b):
         """
         Unites ``GeometryBuilder``.
@@ -81,6 +96,25 @@ class CellOptionBuilder:
     suffix: str = ''
     designator: str = ''
 
+    @staticmethod
+    def unbuild(option: inp.cell.CellOption_):
+        """
+        Generates ``CellOptionBuilder`` from ``CellOption_``.
+
+        Parameter:
+            option: ``CellOption_`` to unbuild.
+
+        Returns:
+            ``CellOptionBuilder`` for ``CellOption_``.
+        """
+
+        return CellOptionBuilder(
+            mnemonic=option._KEYWORD,
+            parameter=option.value.to_mcnp(),
+            suffix=option.suffix.to_mcnp() if hasattr(option, 'suffix') else '',
+            designator=option.designator.to_mcnp() if hasattr(option, 'designator') else '',
+        )
+
     def build(self):
         """
         Builds ``CellOptionBuilder`` into ``CellOption_``.
@@ -123,6 +157,39 @@ class CellBuilder:
     density: float | types.Real = None
     atoms_or_grams: bool = True
 
+    def append(self, option: CellOptionBuilder):
+        """
+        Stores ``Option_`` in ``CellBuilder``,
+
+        Parameters:
+            option: ``Option_`` to add.
+        """
+
+        self.options[
+            f'{option.mnemonic}{option.suffix if hasattr(option, "suffix") else ""}:{option.designator if hasattr(option, "designator") else ""}'
+        ] = option
+
+    @staticmethod
+    def unbuild(cell: inp.Cell):
+        """
+        Generates ``CellBuilder`` from ``Cell``.
+
+        Parameter:
+            cell: ``Cell` to unbuild.
+
+        Returns:
+            ``CellBuilder`` for ``Cell``.
+        """
+
+        return CellBuilder(
+            number=cell.number.value,
+            material=cell.material.value,
+            density=math.abs(cell.density.value),
+            atoms_or_grams=cell.density > 0,
+            options={option._KEYWORD: CellOptionBuilder.unbuild(option) for option in cell.options},
+            geometry=GeometryBuilder.unbuild(cell.geometry),
+        )
+
     def build(self):
         """
         Builds ``CellBuilder`` into ``Cell``.
@@ -138,7 +205,7 @@ def build(self):
             if self.density
             else None,
             geometry=self.geometry.build(),
-            options=types.Tuple([option.build() for option in self.options]),
+            options=types.Tuple([option.build() for option in self.options.values()]),
         )
 
 
@@ -160,6 +227,26 @@ class SurfaceBuilder:
     prefix: str = None
     transform: int = None
 
+    @staticmethod
+    def unbuild(surface: inp.Surface):
+        """
+        Generates ``SurfaceBuilder`` from ``Surface``.
+
+        Parameter:
+            surface: ``Surface` to unbuild.
+
+        Returns:
+            ``SurfaceBuilder`` for ``Surface``.
+        """
+
+        return SurfaceBuilder(
+            number=surface.number.value,
+            parameter=surface.option.value.to_mcnp(),
+            transform=surface.transform.value,
+            mnemonic=surface.option._KEYWORD,
+            prefix=surface.prefix,
+        )
+
     def build(self):
         """
         Builds ``SurfaceBuilder`` into ``Surface``.
@@ -204,6 +291,27 @@ class DataBuilder:
     suffix: str = ''
     designator: str = ''
 
+    @staticmethod
+    def unbuild(data: inp.Data):
+        """
+        Generates ``DataBuilder`` from ``Data``.
+
+        Parameter:
+            data: ``Data` to unbuild.
+
+        Returns:
+            ``DataBuilder`` for ``Data``.
+        """
+
+        return DataBuilder(
+            mnemonic=data.option._KEYWORD,
+            parameter=data.option.value.to_mcnp(),
+            suffix=data.option.suffix.to_mcnp() if hasattr(data.option, 'suffix') else '',
+            designator=data.option.designator.to_mcnp()
+            if hasattr(data.option, 'designator')
+            else '',
+        )
+
     def build(self):
         """
         Builds ``DataBuilder`` into ``Data``.
@@ -246,9 +354,9 @@ class InpBuilder:
     """
 
     title: str
-    cells: tuple[CellBuilder] = tuple()
-    surfaces: tuple[SurfaceBuilder] = tuple()
-    data: tuple[DataBuilder] = tuple()
+    cells: dict[str, CellBuilder] = dataclasses.field(default_factory=lambda: ({}))
+    surfaces: dict[str, SurfaceBuilder] = dataclasses.field(default_factory=lambda: ({}))
+    data: dict[str, DataBuilder] = dataclasses.field(default_factory=lambda: ({}))
     message: str = ''
     other: str = ''
 
@@ -261,11 +369,11 @@ def append(self, card: CellBuilder | SurfaceBuilder | DataBuilder):
         """
 
         if isinstance(card, CellBuilder):
-            self.cells = (*self.cells, card)
+            self.cells[card.number] = card
         elif isinstance(card, SurfaceBuilder):
-            self.surfaces = (*self.surfaces, card)
+            self.surfaces[card.number] = card
         elif isinstance(card, DataBuilder):
-            self.data = (*self.data, card)
+            self.data[card.mnemonic + card.suffix if hasattr(card, 'suffix') else ''] = card
 
     def build(self):
         """
@@ -279,10 +387,10 @@ def build(self):
             title=self.title,
             message=self.message,
             other=self.other,
-            cells=types.Tuple([cell.build() for cell in self.cells]),
+            cells=types.Tuple([cell.build() for cell in self.cells.values()]),
             cells_comments=types.Tuple([]),
-            surfaces=types.Tuple([surface.build() for surface in self.surfaces]),
+            surfaces=types.Tuple([surface.build() for surface in self.surfaces.values()]),
             surfaces_comments=types.Tuple([]),
-            data=types.Tuple([data.build() for data in self.data]),
+            data=types.Tuple([data.build() for data in self.data.values()]),
             data_comments=types.Tuple([]),
         )

examples/data/input_files/png_e.i renamed to examples/data/inp/png_e.i

@@ -24,7 +24,7 @@ class Cell(Card_):
     }
 
     _REGEX = re.compile(
-        rf'\A(\S+)( \S+)((?<! 0) \S+|(?<= 0))( [^a-z]+)(( {cell.CellOption_._REGEX.pattern})+)?\Z'
+        rf'\A(\S+)( \S+)((?<! 0) \S+|(?<= 0))( [^a-z]+)( ({cell.CellOption_._REGEX.pattern}))*\Z'
     )
 
     def __init__(