Implements More Tests and Begins Dice Refactor Implementation

2023-04-22 11:27:20 -03:00
parent 71744641ff
commit b869ee74fb
19 changed files with 719 additions and 297 deletions
--- a/tests/shared/external/test_dice.py
+++ b/tests/shared/external/test_dice.py
@@ -0,0 +1,116 @@
+from diceplayer.shared.external.dice import Dice
+
+from unittest import mock
+import unittest
+
+
+class TestDice(unittest.TestCase):
+    def test_class_instantiation(self):
+        dice = Dice(
+            {
+                'ljname': 'test',
+                'outname': 'test',
+                'ncores': 1,
+                'dens': 1.0,
+                'nmol': [1],
+                'nstep': [1],
+            }
+        )
+
+        self.assertIsInstance(dice, Dice)
+
+    def test_configure(self):
+        dice = Dice(
+            {
+                'ljname': 'test',
+                'outname': 'test',
+                'ncores': 1,
+                'dens': 1.0,
+                'nmol': [1],
+                'nstep': [1],
+            }
+        )
+
+        self.assertFalse(hasattr(dice, 'step'))
+
+        dice.configure('test')
+
+        self.assertTrue(hasattr(dice, 'step'))
+
+    def test_reset(self):
+        dice = Dice(
+            {
+                'ljname': 'test',
+                'outname': 'test',
+                'ncores': 1,
+                'dens': 1.0,
+                'nmol': [1],
+                'nstep': [1],
+            }
+        )
+
+        dice.configure('test')
+
+        self.assertTrue(hasattr(dice, 'step'))
+
+        dice.reset()
+
+        self.assertFalse(hasattr(dice, 'step'))
+
+    @mock.patch('diceplayer.shared.external.dice.connection')
+    @mock.patch('diceplayer.shared.external.dice.Process')
+    def test_start(self, mock_process, mock_connection):
+        dice = Dice(
+            {
+                'ljname': 'test',
+                'outname': 'test',
+                'ncores': 1,
+                'dens': 1.0,
+                'nmol': [1],
+                'nstep': [1],
+            }
+        )
+        dice.start(1)
+
+        self.assertTrue(mock_process.called)
+        self.assertTrue(mock_connection.wait.called)
+
+    def test_simulation_process_raises_exception(self):
+        dice = Dice(
+            {
+                'ljname': 'test',
+                'outname': 'test',
+                'ncores': 1,
+                'dens': 1.0,
+                'nmol': [1],
+                'nstep': [1],
+            }
+        )
+
+        with self.assertRaises(SystemExit):
+            dice._simulation_process(1, 1)
+
+    @mock.patch('diceplayer.shared.external.dice.Dice._make_proc_dir')
+    @mock.patch('diceplayer.shared.external.dice.Dice._make_dice_inputs')
+    @mock.patch('diceplayer.shared.external.dice.Dice._run_dice')
+    def test_simulation_process(self, mock_run_dice, mock_make_dice_inputs, mock_make_proc_dir):
+        dice = Dice(
+            {
+                'ljname': 'test',
+                'outname': 'test',
+                'ncores': 1,
+                'dens': 1.0,
+                'nmol': [1],
+                'nstep': [1],
+            }
+        )
+
+        dice._simulation_process(1, 1)
+
+        self.assertTrue(dice._make_proc_dir.called)
+        self.assertTrue(dice._make_dice_inputs.called)
+        self.assertTrue(dice._run_dice.called)
+
+
+if __name__ == '__main__':
+    unittest.main()