Tally segment parsing bug fix.

doc/source/changelog.rst

@@ -5,6 +5,7 @@ MontePy Changelog
 ----------------------
 
 **Bug fixes**
+* Fixed bug with parsing tally segments (#377)
 
 **CI/CD**
 * Fixed project metadata for author to show up correctly on PyPI (#408)

montepy/data_inputs/data_parser.py

@@ -7,6 +7,7 @@
     material,
     mode,
     tally,
+    tally_segment,
     tally_multiplier,
     thermal_scattering,
     universe_input,
@@ -23,6 +24,7 @@
     mode.Mode,
     tally.Tally,
     tally_multiplier.TallyMultiplier,
+    tally_segment.TallySegment,
     thermal_scattering.ThermalScatteringLaw,
     transform.Transform,
     volume.Volume,

montepy/data_inputs/tally_segment.py

@@ -0,0 +1,22 @@
+# Copyright 2024, Battelle Energy Alliance, LLC All Rights Reserved.
+import montepy
+from montepy.data_inputs.data_input import DataInputAbstract
+from montepy.input_parser.tally_parser import TallyParser
+
+
+class TallySegment(DataInputAbstract):
+    """ """
+
+    _parser = TallyParser()
+
+    @staticmethod
+    def _class_prefix():
+        return "fs"
+
+    @staticmethod
+    def _has_number():
+        return True
+
+    @staticmethod
+    def _has_classifier():
+        return 0

tests/test_tally.py

@@ -6,6 +6,7 @@
 
 
 from unittest import TestCase
+import pytest
 
 
 class TestTallyParser(TestCase):
@@ -38,3 +39,9 @@ def test_parsing_tally_multiplier(self):
             print(test)
             input = Input([test], BlockType.DATA)
             data = parse_data(input)
+
+
+@pytest.mark.parametrize("line", ["fs14 -123", "fs12 -456 t"])
+def test_tally_segment_init(line):
+    input = Input([line], BlockType.DATA)
+    data = parse_data(input)