pymcnp/inp: added metaprogrammed tests

Author: BitterB0NG0

src/pymcnp/_text.py scripts/inp_code.pysrc/pymcnp/_text.py renamed to scripts/inp_code.py

@@ -1,7 +1,7 @@
 import re
 import pathlib
 
-import _data
+import inp_data
 
 
 # UTILITIES #
@@ -30,11 +30,6 @@ def SNAKE(name: str) -> str:
     return name.lower()
 
 
-def FILE(name: str) -> str:
-    name = re.sub('/', '_', name)
-    return name.lower()
-
-
 # ATTRIBUTE FUNCTIONS #
 def ATTRS_REGEX(element):
     def _REGEX(element):
@@ -261,5 +256,5 @@ def build_element(element, parent_name, path_dir, depth):
             file.write(ELEMENT(element, parent_name, depth - 1))
 
 
-for card in _data.cards.options:
-    build_element(card, '', pathlib.Path(__file__).parent / 'inp', 3)
+for card in inp_data.cards.options:
+    build_element(card, '', pathlib.Path(__file__).parent.parent / 'src/pymcnp/inp', 3)

src/pymcnp/_data.py scripts/inp_data.pysrc/pymcnp/_data.py renamed to scripts/inp_data.py

@@ -202,7 +202,7 @@ def __init__(
                         ),
                         AttributeScheme(
                             name='stretch',
-                            type='str',
+                            type='types.String',
                             description='Cell exponential transform stretching specifier',
                         ),
                     ),

scripts/inp_tests.py

@@ -0,0 +1,99 @@
+import re
+import pathlib
+import itertools
+
+import inp_data
+
+
+def CAMEL(name: str) -> str:
+    name = re.sub('/', '_', name)
+    if name:
+        return name[0].upper() + name[1:]
+    else:
+        return ''
+
+
+def SNAKE(name: str) -> str:
+    name = re.sub('/', '_', name)
+    return name.lower()
+
+
+def TEST(element, mod, parent_name):
+    return f'''
+class Test_{CAMEL(parent_name)}{CAMEL(element.name)}:
+    """
+    Tests ``{CAMEL(parent_name)}{CAMEL(element.name)}``.
+    """
+
+    class Test_FromMcnp(_Test_FromMcnp):
+        """
+        Tests ``{CAMEL(parent_name)}{CAMEL(element.name)}.from_mcnp``.
+        """
+
+        element = pymcnp.inp.{f"{mod}." if mod else ""}{CAMEL(element.name)}
+        EXAMPLES_VALID = []
+        EXAMPLES_INVALID = []
+'''[1:]
+
+
+path = pathlib.Path(__file__).parent / '..' / 'tests'
+
+
+def get_tests(element, mod, parent_name):
+    if element.options:
+        return [TEST(element, mod, parent_name)] + list(
+            itertools.chain(
+                *(
+                    get_tests(
+                        option, mod + ('.' if mod else '') + SNAKE(element.name), element.name
+                    )
+                    for option in element.options
+                )
+            )
+        )
+    else:
+        return [TEST(element, mod, parent_name)]
+
+
+for card in inp_data.cards.options:
+    tests = get_tests(card, '', '')
+
+    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)
+
+
+{'\n\n'.join(tests)}
+'''[1:]
+
+    path_file = pathlib.Path(__file__).parent.parent / 'tests' / f'test_{card.name}.py'
+    with path_file.open('w') as file:
+        file.write(res)