🐛 refactor wind_transfer calls in TransferFunctionTestGradient for compatibility with python 3.11-12 (#586)

* 🐛 refactor wind_transfer calls in TransferFunctionTestGradient for compatibility with python 3.11-12

* 🐛 fix variable name from 'result' to 'results' for consistency in TransferFunctionTestGradient

Author: selipot

tests/transfer_test.py

@@ -612,48 +612,91 @@ def test_gradient_ekman_case(self):
 
         for i in range(len(self.delta)):
             for j in range(len(self.bld)):
-                wind_transfer_init[i, j], dG_ddelta[i, j], dG_dbld[i, j] = (
-                    wind_transfer(
-                        omega=omega,
-                        z=z,
-                        cor_freq=cor_freq,
-                        delta=self.delta[i],
-                        mu=mu,
-                        bld=self.bld[j],
-                    )
+                results = wind_transfer(
+                    omega=omega,
+                    z=z,
+                    cor_freq=cor_freq,
+                    delta=self.delta[i],
+                    mu=mu,
+                    bld=self.bld[j],
+                )
+                wind_transfer_init[i, j] = (
+                    results[0].item()
+                    if isinstance(results[0], np.ndarray)
+                    else results[0]
                 )
-                wind_transfer_ddelta_plus[i, j], _, _ = wind_transfer(
+                dG_ddelta[i, j] = (
+                    results[1].item()
+                    if isinstance(results[1], np.ndarray)
+                    else results[1]
+                )
+                dG_dbld[i, j] = (
+                    results[2].item()
+                    if isinstance(results[2], np.ndarray)
+                    else results[2]
+                )
+                # wind_transfer_init[i, j], dG_ddelta[i, j], dG_dbld[i, j] = (
+                #     wind_transfer(
+                #         omega=omega,
+                #         z=z,
+                #         cor_freq=cor_freq,
+                #         delta=self.delta[i],
+                #         mu=mu,
+                #         bld=self.bld[j],
+                #     )
+                # )
+                results = wind_transfer(
                     omega=omega,
                     z=z,
                     cor_freq=cor_freq,
                     delta=self.delta[i] + delta_delta / 2,
                     mu=mu,
                     bld=self.bld[j],
                 )
-                wind_transfer_ddelta_minus[i, j], _, _ = wind_transfer(
+                wind_transfer_ddelta_plus[i, j] = (
+                    results[0].item()
+                    if isinstance(results[0], np.ndarray)
+                    else results[0]
+                )
+                results = wind_transfer(
                     omega=omega,
                     z=z,
                     cor_freq=cor_freq,
                     delta=self.delta[i] - delta_delta / 2,
                     mu=mu,
                     bld=self.bld[j],
                 )
-                wind_transfer_dbld_plus[i, j], _, _ = wind_transfer(
+                wind_transfer_ddelta_minus[i, j] = (
+                    results[0].item()
+                    if isinstance(results[0], np.ndarray)
+                    else results[0]
+                )
+                results = wind_transfer(
                     omega=omega,
                     z=z,
                     cor_freq=cor_freq,
                     delta=self.delta[i],
                     mu=mu,
                     bld=self.bld[j] + delta_bld / 2,
                 )
-                wind_transfer_dbld_minus[i, j], _, _ = wind_transfer(
+                wind_transfer_dbld_plus[i, j] = (
+                    results[0].item()
+                    if isinstance(results[0], np.ndarray)
+                    else results[0]
+                )
+                results = wind_transfer(
                     omega=omega,
                     z=z,
                     cor_freq=cor_freq,
                     delta=self.delta[i],
                     mu=mu,
                     bld=self.bld[j] - delta_bld / 2,
                 )
+                wind_transfer_dbld_minus[i, j] = (
+                    results[0].item()
+                    if isinstance(results[0], np.ndarray)
+                    else results[0]
+                )
 
         dG_ddelta_fd = (
             wind_transfer_ddelta_plus - wind_transfer_ddelta_minus