Implements Refactoring in Player Class and Initial Working Version

tests/shared/interface/test_gaussian_interface.py

@@ -0,0 +1,113 @@
+from diceplayer.shared.interface.gaussian_interface import GaussianInterface
+from diceplayer.shared.config.player_config import PlayerConfig
+from diceplayer.shared.environment.system import System
+from diceplayer import logger
+
+from tests.mocks.mock_inputs import get_config_example
+
+import yaml
+import io
+
+
+from unittest import mock
+import unittest
+
+
+class TestGaussianInterface(unittest.TestCase):
+    def setUp(self) -> None:
+        logger.set_logger(stream=io.StringIO())
+
+        config = yaml.load(get_config_example(), Loader=yaml.Loader)
+        self.config = PlayerConfig.from_dict(config['diceplayer'])
+
+    def test_class_instantiation(self):
+        gaussian_interface = GaussianInterface()
+        self.assertIsInstance(gaussian_interface, GaussianInterface)
+
+    def test_configure(self):
+        gaussian_interface = GaussianInterface()
+
+        self.assertIsNone(gaussian_interface.step)
+        self.assertIsNone(gaussian_interface.system)
+
+        gaussian_interface.configure(self.config, System())
+
+        self.assertIsNotNone(gaussian_interface.step)
+        self.assertIsNotNone(gaussian_interface.system)
+
+    def test_reset(self):
+        gaussian_interface = GaussianInterface()
+
+        gaussian_interface.configure(self.config, System())
+
+        self.assertIsNotNone(gaussian_interface.step)
+        self.assertIsNotNone(gaussian_interface.system)
+
+        gaussian_interface.reset()
+
+        self.assertFalse(hasattr(gaussian_interface, 'step'))
+        self.assertFalse(hasattr(gaussian_interface, 'system'))
+
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.Path.mkdir')
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.Path.exists')
+    def test_make_qm_dir(self, mock_exists, mock_mkdir):
+        mock_exists.return_value = False
+
+        gaussian_interface = GaussianInterface()
+        gaussian_interface.configure(self.config, System())
+
+        gaussian_interface._make_qm_dir(1)
+
+        mock_exists.assert_called_once()
+        mock_mkdir.assert_called_once()
+
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.shutil.copy')
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.Path.exists')
+    def test_copy_chk_file_from_previous_step(self, mock_exists, mock_copy):
+        gaussian_interface = GaussianInterface()
+        gaussian_interface.configure(self.config, System())
+
+        mock_exists.side_effect = [False, True]
+
+        gaussian_interface._copy_chk_file_from_previous_step(2)
+
+        self.assertTrue(mock_exists.called)
+        self.assertTrue(mock_copy.called)
+
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.shutil.copy')
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.Path.exists')
+    def test_copy_chk_file_from_previous_step_no_previous_step(self, mock_exists, mock_copy):
+        gaussian_interface = GaussianInterface()
+        gaussian_interface.configure(self.config, System())
+
+        mock_exists.side_effect = [False, False]
+
+        with self.assertRaises(FileNotFoundError):
+            gaussian_interface._copy_chk_file_from_previous_step(2)
+
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.shutil.copy')
+    @mock.patch('diceplayer.shared.interface.gaussian_interface.Path.exists')
+    def test_copy_chk_file_from_previous_step_current_exists(self, mock_exists, mock_copy):
+        gaussian_interface = GaussianInterface()
+        gaussian_interface.configure(self.config, System())
+
+        mock_exists.side_effect = [True, True]
+
+        with self.assertRaises(FileExistsError):
+            gaussian_interface._copy_chk_file_from_previous_step(2)
+
+    # def test_start(self):
+    #     gaussian_interface = GaussianInterface()
+    #     gaussian_interface.configure(self.config, System())
+    #
+    #     gaussian_interface._make_qm_dir = mock.Mock()
+    #     gaussian_interface._copy_chk_file_from_previous_step = mock.Mock()
+    #
+    #     gaussian_interface.start(2)
+    #
+    #     gaussian_interface._make_qm_dir.assert_called_once_with(2)
+    #     gaussian_interface._copy_chk_file_from_previous_step.assert_called_once_with(2)
+
+
+if __name__ == '__main__':
+    unittest.main()