types: fixed coverage

Author: BitterB0NG0

files/outp/valid_04.outp files/outp/todo_04.outpfiles/outp/valid_04.outp renamed to files/outp/todo_04.outp

@@ -6188,7 +6188,7 @@ def from_formula(number: int, formulas: dict[str, float], is_weight: bool = True
                             ],
                         ),
                         ElementScheme(
-                            name='tme_0',
+                            name='tme_1',
                             mnemonic='tme',
                             attributes=[
                                 AttributeScheme(

files/outp/valid_05.outp files/outp/todo_05.outpfiles/outp/valid_05.outp renamed to files/outp/todo_05.outp

@@ -16,7 +16,9 @@ class Ptrac(_object.McnpFile):
         histories: PTRAC histories.
     """
 
-    _REGEX = re.compile(r'\A(   -1\n.{60}\n[^\n]{80}\n(?:\s[^\n]{120}\n)+\s[^\n]{100}\n(?:\s(?:[^\n]{4})+\n)+?)((?:[^\n]+\n(?:(?:(?:(?:\s[^\n]{50})|(?:\s[^\n]{60})|(?:\s[^\n]{60})|(?:\s[^\n]{70})|(?:\s[^\n]{60})|(?:\s[^\n]{70})|(?:\s[^\n]{70})|(?:\s[^\n]{80}))\n(?:\s[^\n]{39})|(?:\s[^\n]{117}))\n)+)+)\Z')
+    _REGEX = re.compile(
+        r'\A(   -1\n.{60}\n[^\n]{80}\n(?:\s[^\n]{120}\n)+\s[^\n]{100}\n(?:\s(?:[^\n]{4})+\n)+?)((?:[^\n]+\n(?:(?:(?:(?:\s[^\n]{50})|(?:\s[^\n]{60})|(?:\s[^\n]{60})|(?:\s[^\n]{70})|(?:\s[^\n]{60})|(?:\s[^\n]{70})|(?:\s[^\n]{70})|(?:\s[^\n]{80}))\n(?:\s[^\n]{39})|(?:\s[^\n]{117}))\n)+)+)\Z'
+    )
 
     def __init__(self, header: ptrac.Header, histories: typing.Generator[ptrac.History, None, None]):
         """

scripts/inp_data.py

@@ -1,4 +1,4 @@
-from . import _card
+from ._card import Card
 from ._option import Option
 from . import cell
 from . import like

src/pymcnp/Ptrac.py

@@ -1,7 +1,7 @@
 import re
 
+from . import tme_1
 from . import _option
-from .... import types
 from .... import errors
 
 
@@ -13,12 +13,12 @@ class Tme_1(_option.SdefOption):
     _KEYWORD = 'tme'
 
     _ATTRS = {
-        'time': types.EmbeddedDistribution,
+        'time': tme_1.Embedded,
     }
 
-    _REGEX = re.compile(rf'\Atme( {types.EmbeddedDistribution._REGEX.pattern[2:-2]})\Z')
+    _REGEX = re.compile(rf'\Atme( {tme_1.Embedded._REGEX.pattern[2:-2]})\Z')
 
-    def __init__(self, time: str | types.EmbeddedDistribution):
+    def __init__(self, time: str | tme_1.Embedded):
         """
         Initializes ``Tme_1``.
 
@@ -29,10 +29,10 @@ def __init__(self, time: str | types.EmbeddedDistribution):
             InpError: SEMANTICS_OPTION.
         """
 
-        self.time: types.EmbeddedDistribution = time
+        self.time: tme_1.Embedded = time
 
     @property
-    def time(self) -> types.EmbeddedDistribution:
+    def time(self) -> tme_1.Embedded:
         """
         Time in shakes
 
@@ -44,7 +44,7 @@ def time(self) -> types.EmbeddedDistribution:
         return self._time
 
     @time.setter
-    def time(self, time: str | types.EmbeddedDistribution) -> None:
+    def time(self, time: str | tme_1.Embedded) -> None:
         """
         Sets ``time``.
 
@@ -57,12 +57,12 @@ def time(self, time: str | types.EmbeddedDistribution) -> None:
         """
 
         if time is not None:
-            if isinstance(time, types.EmbeddedDistribution):
+            if isinstance(time, tme_1.Embedded):
                 time = time
             elif isinstance(time, str):
-                time = types.EmbeddedDistribution.from_mcnp(time)
+                time = tme_1.Embedded.from_mcnp(time)
 
         if time is None:
             raise errors.InpError(errors.InpCode.SEMANTICS_OPTION, time)
 
-        self._time: types.EmbeddedDistribution = time
+        self._time: tme_1.Embedded = time

src/pymcnp/inp/__init__.py

@@ -1,5 +1,6 @@
 from ._option import SdefOption
 
+from . import tme_1
 from .Cel import Cel
 from .Sur import Sur
 from .Erg_0 import Erg_0
@@ -32,6 +33,7 @@
 
 __all__ = [
     'SdefOption',
+    'tme_1',
     'Cel',
     'Sur',
     'Erg_0',

src/pymcnp/inp/data/sdef/Tme_1.py

@@ -0,0 +1,66 @@
+import re
+
+from . import _entry
+from ..... import types
+from ..... import errors
+
+
+class Embedded(_entry.TmeEntry_1):
+    """
+    Represents INP embedded distribution entries.
+
+    Attributes:
+        distributions: Distribution distributions.
+    """
+
+    _REGEX = re.compile(rf'\A({types.Distribution._REGEX.pattern[2:-2]}(?: ?< ?{types.Distribution._REGEX.pattern[2:-2]})*)\Z')
+
+    def __init__(self, distributions: types.Tuple[types.Distribution]):
+        """
+        Initializes ``Embedded``.
+
+        Parameters:
+            distributions: Time distributions.
+
+        Raises:
+            TypesError: SEMANTICS_TYPE.
+        """
+
+        if distributions is None or None in distributions:
+            raise errors.TypesError(errors.TypesCode.SEMANTICS_TYPE, distributions)
+
+        self.distributions: types.Tuple[types.Distribution] = distributions
+
+    @staticmethod
+    def from_mcnp(source: str):
+        """
+        Generates ``Embedded`` from MCNP.
+
+        Parameters:
+            source: MCNP for ``Embedded``.
+
+        Returns:
+            ``Embedded`` object.
+
+        Raises:
+            TypesError: SYNTAX_TYPE.
+        """
+
+        tokens = Embedded._REGEX.match(source)
+
+        if not tokens:
+            raise errors.TypesError(errors.TypesCode.SYNTAX_TYPE, source)
+
+        distributions = types.Tuple([types.Distribution.from_mcnp(token.strip()) for token in source.split('<')])
+
+        return Embedded(distributions)
+
+    def to_mcnp(self):
+        """
+        Generates MCNP from ``Embedded``.
+
+        Returns:
+            MCNP for ``Embedded``.
+        """
+
+        return '<'.join(number.to_mcnp() for number in self.distributions)

src/pymcnp/inp/data/sdef/__init__.py

@@ -0,0 +1,7 @@
+from ._entry import TmeEntry_1
+from .Embedded import Embedded
+
+__all__ = [
+    'TmeEntry_1',
+    'Embedded',
+]

src/pymcnp/inp/data/sdef/tme_1/Embedded.py

@@ -0,0 +1,9 @@
+from .... import _entry
+
+
+class TmeEntry_1(_entry.Entry):
+    """
+    Represents generic INP tme entries.
+    """
+
+    pass