Refactor tests for type stability

Author: tomasaschan

test/extrapolation/type-stability.jl

@@ -8,26 +8,22 @@ xmax = 10
 A = Float64[f(x) for x in 1:xmax]
 itpg = interpolate(A, BSpline(Linear), OnGrid)
 
-schemes = (Flat, Linear, Reflect, Periodic)
-
-
-for E in schemes
-    etp1 = extrapolate(itpg, E)
-    println(E)
-    
-    # In-bounds evaluation
-    @inferred(getindex(etp1, 3.4))
-    @inferred(getindex(etp1, 3))
-    @inferred(getindex(etp1, dual(3,1)))
-
-    # Out-of-bounds evaluation
-    @inferred(getindex(etp1, -3.4))
-    @inferred(getindex(etp1, -3))
-    @inferred(getindex(etp1, dual(-3,1)))
-
-    @inferred(getindex(etp1, 13.4))
-    @inferred(getindex(etp1, 13))
-    @inferred(getindex(etp1, dual(13,1)))
+schemes = (
+    Flat,
+    Linear,
+    Reflect,
+    Periodic
+)
+
+for etp in map(E -> extrapolate(itpg, E), schemes),
+    x in [
+        # In-bounds evaluation
+        3.4, 3, dual(3.1),
+        # Out-of-bounds evaluation
+        -3.4, -3, dual(-3,1),
+        13.4, 13, dual(13,1)
+    ]
+    @inferred(getindex(etp, x))
 end
 
 # Test type-stability of 2-dimensional extrapolation with homogeneous scheme
@@ -36,33 +32,23 @@ ymax = 4
 A = Float64[f(x)*g(y) for x in 1:xmax, y in 1:ymax]
 itp2 = interpolate(A, BSpline(Linear), OnGrid)
 
-for E in schemes
-    etp2 = extrapolate(itp2, E)
-
-    # In-bounds evaluation
-    @inferred(getindex(etp2, 3.4, 2.1))
-    @inferred(getindex(etp2, 3,2))
-    @inferred(getindex(etp2, dual(3,1), 2.3))
-
-    # Out-of-bounds evaluation
-    @inferred(getindex(etp2, -3.4, 2.1))
-    @inferred(getindex(etp2, 3.4, -2.1))
-    @inferred(getindex(etp2, -3, 2))
-    @inferred(getindex(etp2, 3, -2))
-    @inferred(getindex(etp2, dual(-3,1), 2.3))
-    @inferred(getindex(etp2, dual(3,1), -2.3))
-    @inferred(getindex(etp2, -3.1, dual(-2.3,0)))
-    @inferred(getindex(etp2, 3,1, dual(-2.3,1)))
-
-    @inferred(getindex(etp2, 13.4,2.1))
-    @inferred(getindex(etp2, 13,2))
-    @inferred(getindex(etp2, dual(13,1),2.1))
-    @inferred(getindex(etp2, 3.4,12.1))
-    @inferred(getindex(etp2, 3,12))
-    @inferred(getindex(etp2, dual(3,1),12.1))
-    @inferred(getindex(etp2, dual(13,1),2.1))
-    @inferred(getindex(etp2, 3.1,dual(12.1,1)))
-    @inferred(getindex(etp2, 13,dual(12,1)))
+for (etp2,E) in map(E -> (extrapolate(itp2, E), E), schemes),
+    x in (
+        # In-bounds evaluation
+        3.4, 3, dual(3.1),
+        # Out-of-bounds evaluation
+        -3.4, -3, dual(-3,1),
+        13.4, 13, dual(13,1)
+    ),
+    y in (
+        # In-bounds evaluation
+        2.1, 2, dual(2.3, 1),
+        # Out-of-bounds evaluation
+        -2.1, -2, dual(-2.3, 1),
+        12.1, 12, dual(12.1, 1)
+    )
+
+    @inferred(getindex(etp2, x, y))
 end
 
 end