feat: improved implementation and validations of configs

2026-02-26 08:35:59 -03:00
parent e5c6282c86
commit c51d07cff2
11 changed files with 84 additions and 57 deletions
--- a/tests/config/test_dice_dto.py
+++ b/tests/config/test_dice_dto.py
@@ -0,0 +1,96 @@
+from diceplayer.config.dice_config import DiceConfig
+
+import unittest
+
+
+class TestDiceDto(unittest.TestCase):
+    def test_class_instantiation(self):
+        dice_dto = DiceConfig(
+            ljname="test",
+            outname="test",
+            dens=1.0,
+            nmol=[1],
+            nstep=[1, 1],
+        )
+
+        self.assertIsInstance(dice_dto, DiceConfig)
+
+    def test_validate_jname(self):
+        with self.assertRaises(ValueError) as ex:
+            DiceConfig(
+                ljname=None,
+                outname="test",
+                dens=1.0,
+                nmol=[1],
+                nstep=[1, 1],
+            )
+            self.assertEqual(
+                ex.exception, "Error: 'ljname' keyword not specified in config file"
+            )
+
+    def test_validate_outname(self):
+        with self.assertRaises(ValueError) as ex:
+            DiceConfig(
+                ljname="test",
+                outname=None,
+                dens=1.0,
+                nmol=[1],
+                nstep=[1, 1],
+            )
+            self.assertEqual(
+                ex.exception, "Error: 'outname' keyword not specified in config file"
+            )
+
+    def test_validate_dens(self):
+        with self.assertRaises(ValueError) as ex:
+            DiceConfig(
+                ljname="test",
+                outname="test",
+                dens=None,
+                nmol=[1],
+                nstep=[1, 1],
+            )
+            self.assertEqual(
+                ex.exception, "Error: 'dens' keyword not specified in config file"
+            )
+
+    def test_validate_nmol(self):
+        with self.assertRaises(ValueError) as ex:
+            DiceConfig(
+                ljname="test",
+                outname="test",
+                dens=1.0,
+                nmol=0,
+                nstep=[1, 1],
+            )
+            self.assertEqual(
+                ex.exception,
+                "Error: 'nmol' keyword not defined appropriately in config file",
+            )
+
+    def test_validate_nstep(self):
+        with self.assertRaises(ValueError) as ex:
+            DiceConfig(
+                ljname="test",
+                outname="test",
+                dens=1.0,
+                nmol=[1],
+                nstep=0,
+            )
+            self.assertEqual(
+                ex.exception,
+                "Error: 'nstep' keyword not defined appropriately in config file",
+            )
+
+    def test_from_dict(self):
+        dice_dto = DiceConfig.model_validate(
+            {
+                "ljname": "test",
+                "outname": "test",
+                "dens": 1.0,
+                "nmol": [1],
+                "nstep": [1, 1],
+            }
+        )
+
+        self.assertIsInstance(dice_dto, DiceConfig)