add tests or new methods on aquifer

    aquifer = Aquifer(**base_params)

    assert aquifer.thickness == 10.0

    assert aquifer.porosity == 0.2

def test_transmissivity_calculated_from_permeability(base_params):

    aquifer = Aquifer(**base_params)

    assert aquifer.transmissivity == base_params["permeability"] * base_params["thickness"]

def test_default_mask_is_nan(base_params):

    aquifer = Aquifer(**base_params)

    assert np.isnan(aquifer.mask)

def test_mask_is_set_when_provided(base_params):

    base_params["mask"] = xr.DataArray(np.array([np.nan, 1.0]))

    aquifer = Aquifer(**base_params)

    assert not np.isnan(aquifer.mask.values[1])

def test_calculate_temperature(base_params):

    aquifer = Aquifer(**base_params)

    gradient, surface_temp = 31.0, 10.0

    aquifer.calculate_temperature(gradient, surface_temp)

    expected = surface_temp + (base_params["depth"] + base_params["thickness"] / 2) * (gradient * 0.001)

    assert aquifer.temperature == pytest.approx(expected)

 No newline at end of file

from pytg3.aquifer import Aquifer

from pytg3.doublet import ThermoGISDoublet

import xarray as xr

def test_deterministic_doublet():

    aquifer = Aquifer(

        thickness=10,

        porosity=0.2,

        ntg=0.5,

        depth=2000,

        permeability=200,

    )

    doublet = ThermoGISDoublet(aquifer=aquifer)

    results1 = doublet.simulate()

    results2 = doublet.simulate()

    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