From 842d3701ff7baab236922200deaf6a23da31fb71 Mon Sep 17 00:00:00 2001
From: Santiago Soler <santisoler@fastmail.com>
Date: Wed, 13 Sep 2023 14:58:44 -0700
Subject: [PATCH] Specify nopython=True on jit functions

Explicitly set `nopython=True` on jit function in tests and examples.
This avoids some warnings about the change of the default configuration
in Numba: by default `nopython` will be set to `True` in the future.
---
 harmonica/_forward/tesseroid.py                    |  2 +-
 harmonica/tests/test_tesseroid_variable_density.py | 14 +++++++-------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/harmonica/_forward/tesseroid.py b/harmonica/_forward/tesseroid.py
index 481faa32d..b8ca19c18 100644
--- a/harmonica/_forward/tesseroid.py
+++ b/harmonica/_forward/tesseroid.py
@@ -139,7 +139,7 @@ def tesseroid_gravity(
     >>> # Define a linear density function for the same tesseroid.
     >>> # It should be decorated with numba.njit
     >>> from numba import jit
-    >>> @jit
+    >>> @jit(nopython=True)
     ... def linear_density(radius):
     ...     density_top = 2670.
     ...     density_bottom = 3300.
diff --git a/harmonica/tests/test_tesseroid_variable_density.py b/harmonica/tests/test_tesseroid_variable_density.py
index 6c20ae934..0c1da5e6b 100644
--- a/harmonica/tests/test_tesseroid_variable_density.py
+++ b/harmonica/tests/test_tesseroid_variable_density.py
@@ -95,7 +95,7 @@ def fixture_quadratic_density(quadratic_params):
     """
     factor, vertex_radius, vertex_density = quadratic_params
 
-    @jit
+    @jit(nopython=True)
     def density(radius):
         """Quadratic density function"""
         return factor * (radius - vertex_radius) ** 2 + vertex_density
@@ -111,7 +111,7 @@ def fixture_straight_line_analytic(bottom, top, quadratic_density):
     density_bottom, density_top = quadratic_density(bottom), quadratic_density(top)
     slope = (density_top - density_bottom) / (top - bottom)
 
-    @jit
+    @jit(nopython=True)
     def line(radius):
         return slope * (radius - bottom) + density_bottom
 
@@ -197,7 +197,7 @@ def test_density_minmax_exponential_function(bottom, top):
     thickness = top - bottom
     b_factor = 50
 
-    @jit
+    @jit(nopython=True)
     def exponential_density(radius):
         """
         Create a dummy exponential density
@@ -276,7 +276,7 @@ def test_density_based_discret_linear_density():
     w, e, s, n, bottom, top = -3, 2, -4, 5, 30, 50
     tesseroid = [w, e, s, n, bottom, top]
 
-    @jit
+    @jit(nopython=True)
     def linear_density(radius):
         """Define a dummy linear density"""
         return 3 * radius + 2
@@ -323,7 +323,7 @@ def test_single_tesseroid_against_constant_density(field):
     density = 2900.0
 
     # Define a constant density
-    @jit
+    @jit(nopython=True)
     def constant_density(
         radius,  # noqa: U100 # the radius argument is needed for the density function
     ):
@@ -440,7 +440,7 @@ def test_spherical_shell_linear_density(field, thickness):
     slope = (density_outer - density_inner) / (top - bottom)
     constant_term = density_outer - slope * top
 
-    @jit
+    @jit(nopython=True)
     def linear_density(radius):
         """
         Create a dummy linear density
@@ -480,7 +480,7 @@ def test_spherical_shell_exponential_density(field, thickness, b_factor):
     a_factor = (density_inner - density_outer) / (1 - np.exp(-b_factor))
     constant_term = density_inner - a_factor
 
-    @jit
+    @jit(nopython=True)
     def exponential_density(radius):
         """
         Create a dummy exponential density