Loading pythermogis/pytg3/doublet.py +1 −2 Original line number Diff line number Diff line Loading @@ -12,13 +12,12 @@ from pydantic import BaseModel, ConfigDict from pythermogis.dask import auto_chunk_dataset from pythermogis.mock import create_logger_mock from pythermogis.pytg3.settings import Settings from pythermogis.pytg3.typing import FloatOrArray logger = logging.getLogger(__name__) if TYPE_CHECKING: from pythermogis.pytg3.aquifer import Aquifer from pythermogis.pytg3.typing import FloatOrArray class ThermoGISDoubletResults(BaseModel): Loading tests/pytg3/test_doublet.py +18 −0 Original line number Diff line number Diff line Loading @@ -19,3 +19,21 @@ def test_deterministic_doublet(): xr.testing.assert_equal(results1.to_dataset(), results2.to_dataset()) def test_deterministic_doublet_mask_value(): aquifer = Aquifer( thickness=10, porosity=0.2, ntg=0.5, depth=2000, permeability=200, mask=0.0 ) doublet = ThermoGISDoublet(aquifer=aquifer) results = doublet.simulate(mask_value=-9999.0) assert results.power == -9999.0 No newline at end of file Loading
pythermogis/pytg3/doublet.py +1 −2 Original line number Diff line number Diff line Loading @@ -12,13 +12,12 @@ from pydantic import BaseModel, ConfigDict from pythermogis.dask import auto_chunk_dataset from pythermogis.mock import create_logger_mock from pythermogis.pytg3.settings import Settings from pythermogis.pytg3.typing import FloatOrArray logger = logging.getLogger(__name__) if TYPE_CHECKING: from pythermogis.pytg3.aquifer import Aquifer from pythermogis.pytg3.typing import FloatOrArray class ThermoGISDoubletResults(BaseModel): Loading
tests/pytg3/test_doublet.py +18 −0 Original line number Diff line number Diff line Loading @@ -19,3 +19,21 @@ def test_deterministic_doublet(): xr.testing.assert_equal(results1.to_dataset(), results2.to_dataset()) def test_deterministic_doublet_mask_value(): aquifer = Aquifer( thickness=10, porosity=0.2, ntg=0.5, depth=2000, permeability=200, mask=0.0 ) doublet = ThermoGISDoublet(aquifer=aquifer) results = doublet.simulate(mask_value=-9999.0) assert results.power == -9999.0 No newline at end of file