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_player_dto.py
+++ b/tests/config/test_player_dto.py
@@ -0,0 +1,83 @@
+from diceplayer.config.dice_config import DiceConfig
+from diceplayer.config.gaussian_config import GaussianConfig
+from diceplayer.config.player_config import PlayerConfig
+
+import unittest
+
+
+def get_config_dict():
+    return {
+        "opt": True,
+        "mem": 12,
+        "maxcyc": 100,
+        "nprocs": 4,
+        "ncores": 4,
+        "dice": {
+            "ljname": "test",
+            "outname": "test",
+            "dens": 1.0,
+            "nmol": [1],
+            "nstep": [1, 1],
+        },
+        "gaussian": {
+            "level": "test",
+            "qmprog": "g16",
+            "keywords": "test",
+        },
+    }
+
+
+class TestPlayerDTO(unittest.TestCase):
+    def setUp(self) -> None:
+        self.dice_dto = DiceConfig(
+            ljname="test",
+            outname="test",
+            dens=1.0,
+            nmol=[1],
+            nstep=[1, 1],
+        )
+        self.gaussian_dto = GaussianConfig(
+            level="test",
+            qmprog="g16",
+            keywords="test",
+        )
+
+    def test_class_instantiation(self):
+        player_dto = PlayerConfig(
+            opt=True,
+            mem=12,
+            maxcyc=100,
+            nprocs=4,
+            ncores=4,
+            dice=self.dice_dto,
+            gaussian=self.gaussian_dto,
+        )
+
+        self.assertIsInstance(player_dto, PlayerConfig)
+        self.assertIsInstance(player_dto.dice, DiceConfig)
+        self.assertIsInstance(player_dto.gaussian, GaussianConfig)
+
+    def test_min_altsteps(self):
+        player_dto = PlayerConfig(
+            opt=True,
+            mem=12,
+            maxcyc=100,
+            nprocs=4,
+            ncores=4,
+            altsteps=100,
+            dice=self.dice_dto,
+            gaussian=self.gaussian_dto,
+        )
+
+        self.assertEqual(player_dto.altsteps, 20000)
+
+    def test_from_dict(self):
+        player_dto = PlayerConfig.model_validate(get_config_dict())
+
+        self.assertIsInstance(player_dto, PlayerConfig)
+        self.assertIsInstance(player_dto.dice, DiceConfig)
+        self.assertIsInstance(player_dto.gaussian, GaussianConfig)
+
+
+if __name__ == "__main__":
+    unittest.main()