refactor: Create a coupler module

Author: phumthep

src/pownet/coupler/__init__.py

@@ -0,0 +1,2 @@
+from .power_water import PowerWaterCoupler
+from .power_power import PowerPowerCoupler

src/pownet/coupler.py src/pownet/coupler/power_water.pysrc/pownet/coupler.py renamed to src/pownet/coupler/power_water.py

@@ -1,10 +1,9 @@
-"""coupler.py: PowerWaterCoupler class to couple the power and water systems."""
+"""power_water.py: PowerWaterCoupler class to couple the power and water systems."""
 
-from .core import ModelBuilder
-from .reservoir.manager import ReservoirManager
+from ..core.model_builder import ModelBuilder
+from ..reservoir.manager import ReservoirManager
 
 import logging
-
 logger = logging.getLogger(__name__)
 
 

src/test_pownet/test_coupler/__init__.py

@@ -1,8 +1,7 @@
-"test_coupler.py"
+"test_power_water.py"
 
 import unittest
 from unittest.mock import MagicMock, PropertyMock, call
-import logging
 
 # Assuming coupler.py is in the same directory or accessible via PYTHONPATH
 from pownet.coupler import PowerWaterCoupler
@@ -11,8 +10,6 @@
 from pownet import ModelBuilder as ActualModelBuilder
 from pownet.reservoir import ReservoirManager as ActualReservoirManager
 
-# Disable logging for cleaner test output
-logging.disable(logging.CRITICAL)
 
 
 class TestPowerWaterCoupler(unittest.TestCase):