shuffle csv, values by index

Author: cbix

tests/test_field.py

@@ -263,14 +263,28 @@ def test_field_interpolation_and_derivatives():
 def test_field_from_csv():
     """Test reading field from csv."""
     data = np.genfromtxt("tests/test_temp.csv", delimiter=",", names=True)
-    dim_time = np.unique(data["time"])
-    dim_axial = np.unique(data["axial"])
-    values = data["Temperature"].reshape(len(dim_time), len(dim_axial))
+
+    # Initialize dimensions
+    dim_time, idx_time = np.unique(data["time"], return_inverse=True)
+    dim_axial, idx_axial = np.unique(data["axial"], return_inverse=True)
+
+    # Initialize values
+    values = np.zeros((len(dim_time), len(dim_axial)))
+    for idx in range(len(data)):
+        values[idx_time[idx]][idx_axial[idx]] = data[idx]["Temperature"]
+
     field = Field(
         name="temp_field",
         dimensions={"time": dim_time, "axial": dim_axial},
         data=values,
     )
+    assert_shape(field.shape, (4, 3), "Field shape mismatch")
+
+    interp_field = FieldInterpolator(field)
+    assert_equal(interp_field(time=1700, axial=0), 24, "Wrong interpolated temperature")
+    assert_equal(
+        interp_field(time=300, axial=0.3), 17.34, "Wrong interpolated temperature"
+    )
 
 
 # %% Run tests

tests/test_temp.csv

@@ -1,11 +1,13 @@
-"time", "axial", "Temperature"
-0, 0.0, 20
-0, 0.5, 20
-0, 1.0, 20
-1800, 0.0, 25
-1800, 0.5, 25
-1800, 1.0, 25
-3600, 0.0, 30
-3600, 0.5, 30
-3600, 1.0, 30
-
+"time","axial","Temperature"
+0,0.0,20
+0,0.5,15
+1500,0.0,22
+1500,0.5,16.5
+1800,0.0,25
+1500,1.0,11
+3600,0.5,22.5
+0,1.0,10
+1800,1.0,12.5
+3600,0.0,30
+1800,0.5,18.75
+3600,1.0,15