Update function to round_and_hash_float_array and add test

Author: VeckoTheGecko

tests/utils.py

@@ -140,7 +140,9 @@ def assert_valid_field_data(data: xr.DataArray, grid: XGrid):
         assert ax_actual == ax_expected, f"Expected axis {ax_expected} for dimension '{dim}', got {ax_actual}"
 
 
-def hash_float_array(arr):
+def round_and_hash_float_array(arr, decimals=6):
+    arr = np.round(arr, decimals=decimals)
+
     # Adapted from https://cs.stackexchange.com/a/37965
     h = 1
     for f in arr:

tests/v4/test_particleset_execute.py

@@ -155,8 +155,8 @@ def test_uxstommelgyre_pset_execute():
         dt=np.timedelta64(60, "s"),
         pyfunc=AdvectionEE,
     )
-    assert utils.hash_float_array([p.lon for p in pset]) == 1165396086
-    assert utils.hash_float_array([p.lat for p in pset]) == 1142124776
+    assert utils.round_and_hash_float_array([p.lon for p in pset]) == 1165396086
+    assert utils.round_and_hash_float_array([p.lat for p in pset]) == 1142124776
 
 
 @pytest.mark.xfail(reason="Output file not implemented yet")

tests/v4/test_utils.py

@@ -0,0 +1,19 @@
+import numpy as np
+
+from tests import utils
+
+
+def test_round_and_hash_float_array():
+    decimals = 7
+    arr = np.array([1.0, 2.0, 3.0], dtype=np.float64)
+    h = utils.round_and_hash_float_array(arr, decimals=decimals)
+    assert h == 1068792616613
+
+    delta = 10**-decimals
+    arr_test = arr + 0.49 * delta
+    h2 = utils.round_and_hash_float_array(arr_test, decimals=decimals)
+    assert h2 == h
+
+    arr_test = arr + 0.51 * delta
+    h3 = utils.round_and_hash_float_array(arr_test, decimals=decimals)
+    assert h3 != h