fix: Update grid size and z-axis limits in periodic spline example

Signed-off-by: Sietze van Buuren <s.van.buuren@gmail.com>

Author: swvanbuuren

cubinterpp/main.py

@@ -132,6 +132,11 @@ def main():
         'data points'
     )
 
+    x, y, f = get_test_data_2d(case='three_bumps')
+    xp = np.repeat(x, y.size)
+    yp = np.tile(y, x.size)
+    zp = f.flatten()
+
     for interp_type in ('linear', 'monotonic', 'akima', 'natural_spline'):
         x_fine, y_fine, z_fine = cubinterpp_interp2(
             interp_type=interp_type,
@@ -146,10 +151,6 @@ def main():
         ax.xticks = 8
         ax.yticks = 8
         ax.zlim = [0, 8]
-        x, y, f = get_test_data_2d(case='three_bumps')
-        xp = np.repeat(x, y.size)
-        yp = np.tile(y, x.size)
-        zp = f.flatten()
         mpg.points3(xp, yp, zp, color=(0.8, 0.1, 0.1, 1), size=3)
 
     x, y, f = get_test_data_2d(case='three_bumps')

cubinterpp/periodic_spline_2d.py

@@ -31,7 +31,7 @@ def main():
     spline_y = cubinterpp.NaturalPeriodicSpline2D(tx, ty, y)
     spline_z = cubinterpp.NaturalPeriodicSpline2D(tx, ty, z)
     # evaluate on finer grid
-    nxi, nyi = 30, 30
+    nxi, nyi = 33, 33
     txi = np.linspace(0.0, 1.0, nxi)
     tyi = np.linspace(0.0, 1.0, nyi)
     x_fine = np.zeros((nxi, nyi))
@@ -50,7 +50,7 @@ def main():
     ax.azimuth = 225
     ax.xlim = [-4, 4]
     ax.ylim = [-4, 4]
-    ax.zlim = [-1.5, 1.5]
+    ax.zlim = [-2.5, 2.5]
     xp = x.flatten()
     yp = y.flatten()
     zp = z.flatten()