Merge pull request #384 from briochemc/master

patch_UniftulVector

Author: mkitti

src/extrapolation/extrapolation.jl

@@ -53,8 +53,7 @@ count_interp_dims(::Type{<:Extrapolation{T,N,ITPT}}, n) where {T,N,ITPT} = count
 end
 @inline function (etp::Extrapolation{T,N})(x::Vararg{Union{Number,AbstractVector},N}) where {T,N}
     itp = parent(etp)
-    Tret = typeof(lispyprod(zero(T), x...))
-    ret = zeros(Tret, shape(x...))
+    ret = zeros(T, shape(x...))
     for (i, y) in zip(eachindex(ret), Iterators.product(x...))
         ret[i] = etp(y...)
     end

test/extrapolation/type-stability.jl

@@ -69,4 +69,10 @@ using Test, Interpolations, DualNumbers, Unitful
     @test @inferred(itp(2018)) === 2.0f0 * u
     @test @inferred(etp(2018)) === 2.0f0 * u
     @test @inferred(etp(2019)) === 1.0f0 * u
+
+    # Unitful vector knots
+    t = [0.0, 1.0] * u"m"
+    itp = LinearInterpolation(t, y)
+    @test itp(0.5u"m") == 1.5f0 * u
+    @test itp([0.5, 0.75]u"m") == [1.5f0, 1.75f0] * u
 end