1+ import logging
12import math
23import os
34import shutil
2021from CodeEntropy .levels import LevelManager
2122from CodeEntropy .main import main
2223from CodeEntropy .run import ConfigManager , RunManager
24+ from tests .test_CodeEntropy .test_base import BaseTestCase
2325
2426
25- class TestEntropyManager (unittest . TestCase ):
27+ class TestEntropyManager (BaseTestCase ):
2628 """
27- Unit tests for the functionality of EntropyManager.
29+ Unit tests for EntropyManager.
2830 """
2931
3032 def setUp (self ):
31- """
32- Set up test environment.
33- """
34- self .test_dir = tempfile .mkdtemp (prefix = "CodeEntropy_" )
33+ super ().setUp ()
3534 self .test_data_dir = os .path .dirname (data .__file__ )
36- self .code_entropy = main
37-
38- # Change to test directory
39- self ._orig_dir = os .getcwd ()
40- os .chdir (self .test_dir )
4135
42- def tearDown (self ):
43- """
44- Clean up after each test.
45- """
46- os .chdir (self ._orig_dir )
47- if os .path .exists (self .test_dir ):
48- shutil .rmtree (self .test_dir )
36+ # Disable MDAnalysis and commands file logging entirely
37+ logging .getLogger ("MDAnalysis" ).handlers = [logging .NullHandler ()]
38+ logging .getLogger ("commands" ).handlers = [logging .NullHandler ()]
4939
5040 def test_execute_full_workflow (self ):
5141 # Setup universe and args
@@ -56,7 +46,7 @@ def test_execute_full_workflow(self):
5646 args = MagicMock (
5747 bin_width = 0.1 , temperature = 300 , selection_string = "all" , water_entropy = False
5848 )
59- run_manager = RunManager ("temp_folder " )
49+ run_manager = RunManager ("mock_folder/job001 " )
6050 level_manager = LevelManager ()
6151 data_logger = DataLogger ()
6252 group_molecules = MagicMock ()
@@ -153,7 +143,7 @@ def test_execute_triggers_handle_water_entropy_minimal(self):
153143 args = MagicMock (
154144 bin_width = 0.1 , temperature = 300 , selection_string = "all" , water_entropy = True
155145 )
156- run_manager = RunManager ("temp_folder " )
146+ run_manager = RunManager ("mock_folder/job001 " )
157147 level_manager = LevelManager ()
158148 data_logger = DataLogger ()
159149 group_molecules = MagicMock ()
@@ -279,7 +269,7 @@ def test_initialize_molecules(self):
279269 args = MagicMock (
280270 bin_width = 0.1 , temperature = 300 , selection_string = "all" , water_entropy = False
281271 )
282- run_manager = RunManager ("temp_folder " )
272+ run_manager = RunManager ("mock_folder/job001 " )
283273 level_manager = LevelManager ()
284274 data_logger = DataLogger ()
285275 group_molecules = MagicMock ()
@@ -486,7 +476,7 @@ def test_get_reduced_universe_reduced(self, mock_args):
486476 u = mda .Universe (tprfile , trrfile )
487477
488478 config_manager = ConfigManager ()
489- run_manager = RunManager ("temp_folder " )
479+ run_manager = RunManager ("mock_folder/job001 " )
490480
491481 parser = config_manager .setup_argparse ()
492482 args = parser .parse_args ()
@@ -524,7 +514,7 @@ def test_get_molecule_container(self, mock_args):
524514
525515 # Setup managers
526516 config_manager = ConfigManager ()
527- run_manager = RunManager ("temp_folder " )
517+ run_manager = RunManager ("mock_folder/job001 " )
528518
529519 parser = config_manager .setup_argparse ()
530520 args = parser .parse_args ()
@@ -639,7 +629,7 @@ def test_process_vibrational_only_levels(self):
639629
640630 # Setup managers and arguments
641631 args = MagicMock (bin_width = 0.1 , temperature = 300 , selection_string = "all" )
642- run_manager = RunManager ("temp_folder " )
632+ run_manager = RunManager ("mock_folder/job001 " )
643633 level_manager = LevelManager ()
644634 data_logger = DataLogger ()
645635 group_molecules = MagicMock ()
@@ -751,7 +741,7 @@ def test_process_conformational_residue_level(self):
751741
752742 # Setup managers and arguments
753743 args = MagicMock (bin_width = 0.1 , temperature = 300 , selection_string = "all" )
754- run_manager = RunManager ("temp_folder " )
744+ run_manager = RunManager ("mock_folder/job001 " )
755745 level_manager = LevelManager ()
756746 data_logger = DataLogger ()
757747 group_molecules = MagicMock ()
@@ -1086,7 +1076,7 @@ def test_vibrational_entropy_init(self):
10861076 args .temperature = 300
10871077 args .selection_string = "all"
10881078
1089- run_manager = RunManager ("temp_folder " )
1079+ run_manager = RunManager ("mock_folder/job001 " )
10901080 level_manager = LevelManager ()
10911081 data_logger = DataLogger ()
10921082 group_molecules = MagicMock ()
@@ -1111,7 +1101,7 @@ def test_frequency_calculation_0(self):
11111101 lambdas = [0 ]
11121102 temp = 298
11131103
1114- run_manager = RunManager ("mock_folder" )
1104+ run_manager = RunManager ("mock_folder/job001 " )
11151105
11161106 ve = VibrationalEntropy (
11171107 run_manager , MagicMock (), MagicMock (), MagicMock (), MagicMock (), MagicMock ()
@@ -1131,7 +1121,7 @@ def test_frequency_calculation_positive(self):
11311121 temp = 298
11321122
11331123 # Create a mock RunManager and set return value for get_KT2J
1134- run_manager = RunManager ("mock_folder" )
1124+ run_manager = RunManager ("mock_folder/job001 " )
11351125
11361126 # Instantiate VibrationalEntropy with mocks
11371127 ve = VibrationalEntropy (
@@ -1273,7 +1263,7 @@ def test_vibrational_entropy_polymer_force(self):
12731263 temp = 298
12741264 highest_level = "yes"
12751265
1276- run_manager = RunManager ("mock_folder" )
1266+ run_manager = RunManager ("mock_folder/job001 " )
12771267 ve = VibrationalEntropy (
12781268 run_manager , MagicMock (), MagicMock (), MagicMock (), MagicMock (), MagicMock ()
12791269 )
@@ -1303,7 +1293,7 @@ def test_vibrational_entropy_polymer_torque(self):
13031293 temp = 298
13041294 highest_level = "yes"
13051295
1306- run_manager = RunManager ("mock_folder" )
1296+ run_manager = RunManager ("mock_folder/job001 " )
13071297 ve = VibrationalEntropy (
13081298 run_manager , MagicMock (), MagicMock (), MagicMock (), MagicMock (), MagicMock ()
13091299 )
@@ -1561,7 +1551,7 @@ def test_confirmational_entropy_init(self):
15611551 args .temperature = 300
15621552 args .selection_string = "all"
15631553
1564- run_manager = RunManager ("temp_folder " )
1554+ run_manager = RunManager ("mock_folder/job001 " )
15651555 level_manager = LevelManager ()
15661556 data_logger = DataLogger ()
15671557 group_molecules = MagicMock ()
@@ -1603,7 +1593,7 @@ def test_assign_conformation(self):
16031593
16041594 # Setup managers and arguments
16051595 args = MagicMock (bin_width = 0.1 , temperature = 300 , selection_string = "all" )
1606- run_manager = RunManager ("temp_folder " )
1596+ run_manager = RunManager ("mock_folder/job001 " )
16071597 level_manager = LevelManager ()
16081598 data_logger = DataLogger ()
16091599 group_molecules = MagicMock ()
@@ -1635,7 +1625,7 @@ def test_conformational_entropy_calculation(self):
16351625
16361626 # Setup managers and arguments
16371627 args = MagicMock (bin_width = 0.1 , temperature = 300 , selection_string = "all" )
1638- run_manager = RunManager ("temp_folder " )
1628+ run_manager = RunManager ("mock_folder/job001 " )
16391629 level_manager = LevelManager ()
16401630 data_logger = DataLogger ()
16411631 group_molecules = MagicMock ()
@@ -1697,7 +1687,7 @@ def test_orientational_entropy_init(self):
16971687 args .temperature = 300
16981688 args .selection_string = "all"
16991689
1700- run_manager = RunManager ("temp_folder " )
1690+ run_manager = RunManager ("mock_folder/job001 " )
17011691 level_manager = LevelManager ()
17021692 data_logger = DataLogger ()
17031693 group_molecules = MagicMock ()
0 commit comments