more testing

Author: daanhb

src/FunctionMaps.jl

@@ -27,7 +27,8 @@ export canonicalmap
 # from generic/composite.jl
 export composedmap
 # from generic/product.jl
-export ProductMap, productmap
+export ProductMap, productmap,
+    factors, nfactors, factor
 
 # from concrete/basic.jl
 export IdentityMap,

src/concrete/affine/special.jl

@@ -1,4 +1,3 @@
-
 ##############################
 # Numbers and arrays as a map
 ##############################

src/generic/product.jl

@@ -1,4 +1,3 @@
-
 """
 A product map is diagonal and acts on each of the components of x separately:
 `y = f(x)` becomes `y_i = f_i(x_i)`.
@@ -85,6 +84,8 @@ canonicalmap(m::ProductMap) = any(map(hascanonicalmap, factors(m))) ?
 	ProductMap(map(canonicalmap, factors(m))) : m
 canonicalmap(::Equal, m::ProductMap) = any(map(hasequalmap, factors(m))) ?
 	ProductMap(map(equalmap, factors(m))) : m
+canonicalmap(::Equivalent, m::ProductMap{NTuple{N,T}}) where {N,T} =
+	equivalentmap(convert(Map{SVector{N,T}}, m))
 canonicalmap(::Equivalent, m::ProductMap) = any(map(hasequivalentmap, factors(m))) ?
 	ProductMap(map(equivalentmap, factors(m))) : m
 

src/util/common.jl

@@ -7,18 +7,20 @@ isrealtype(::Type{Any}) = false
 const StaticTypes = Union{Number,<:StaticVector{N} where N,<:NTuple{N,Any} where N}
 
 "Apply the `hash` function recursively to the given arguments."
+hashrec() = zero(UInt)
 hashrec(x) = hash(x)
 hashrec(x, args...) = hash(x, hashrec(args...))
 
 # Workaround for #88, manually compute the hash of an array using all its elements
-hashrec() = zero(UInt)
-function hashrec(A::AbstractArray, args...)
+function hash_array(A)
 	h = hash(size(A))
 	for x in A
 		h = hash(x, h)
 	end
-	hash(h, hashrec(args...))
+	h
 end
+hashrec(A::AbstractArray) = hash_array(A)
+hashrec(A::AbstractArray, args...) = hash(hash_array(A), hashrec(args...))
 
 "What is the euclidean dimension of the given type (if applicable)?"
 euclideandimension(::Type{T}) where {T <: Number} = 1

test/runtests.jl

@@ -1,9 +1,9 @@
-module FunctionMapsTests
+using Test
 
-using Test, LinearAlgebra, StaticArrays
+using LinearAlgebra, StaticArrays
 using CompositeTypes, CompositeTypes.Indexing
 
-include("using_fmaps.jl")
+using FunctionMaps
 
 include("aqua.jl")
 
@@ -13,8 +13,7 @@ include("test_interface.jl")
 include("test_generic.jl")
 include("test_basic.jl")
 include("test_affine.jl")
+include("test_canonical.jl")
 include("test_product.jl")
 include("test_maps.jl")
 include("test_arithmetics.jl")
-
-end

test/test_affine.jl

@@ -1,3 +1,7 @@
+using FunctionMaps:
+    to_matrix, to_vector,
+    matrix_pinv
+
 function test_affine_maps(T)
     A = rand(T,2,2)
     @test FunctionMaps.to_matrix(Vector{T}, A) == A

test/test_arithmetics.jl

@@ -1,3 +1,7 @@
+using FunctionMaps:
+    interval_map,
+    bounded_interval_map
+
 @testset "map interval" begin
         @test interval_map(1.0, Inf, 2.0, Inf) == AffineMap(1.0, 1.0)
         @test interval_map(1.0, Inf, Inf, 2.0) == AffineMap(-1.0, 3.0)

test/test_canonical.jl

@@ -0,0 +1,30 @@
+using FunctionMaps:
+    hascanonicalmap,
+    hasequalmap,
+    equivalentmap,
+    hasequivalentmap,
+    tofunctionmap,
+    equalmap
+
+struct MyCanonicalType <: FunctionMaps.CanonicalType end
+
+function test_canonical()
+    m = LinearMap(2.0)
+    @test canonicalmap(m) == m
+    @test !hascanonicalmap(m)
+    @test canonicalmap(MyCanonicalType(), m) == m
+    @test !hascanonicalmap(MyCanonicalType(), m)
+
+    @test tofunctionmap(m) == m
+    @test tofunctionmap(MapRef(m)) == m
+
+    @test canonicalmap(FunctionMaps.Equal(), 5.0) isa Map
+    @test hasequalmap(5.0)
+    @test equalmap(5.0) isa ScalarLinearMap{Float64}
+    @test isequalmap(LinearMap(5), 5.0)
+
+    m1 = LinearMap(2.0)
+    m2tuple = FunctionMaps.TupleProductMap(m1, m1)
+    @test hasequivalentmap(m2tuple)
+    @test equivalentmap(m2tuple) isa FunctionMaps.VcatMap
+end

test/test_common.jl

@@ -1,3 +1,24 @@
+
+using FunctionMaps:
+    hashrec,
+    convert_numtype,
+    promote_numtype,
+    to_numtype,
+    convert_prectype,
+    promote_prectype,
+    to_prectype,
+    convert_eltype,
+    euclideandimension,
+    isrealtype
+
+function test_hashrec()
+    @test hashrec() == 0
+    @test hashrec() isa UInt
+    A = [1,2]
+    @test hashrec(A) == hash(2, hash(1, hash((2,))))
+    @test hashrec(1, 2) == hash(1, hash(2))
+end
+
 function test_dimension()
     @test euclideandimension(Int) == 1
     @test euclideandimension(Float64) == 1
@@ -10,6 +31,7 @@ function test_dimension()
 end
 
 function test_prectype()
+    @test prectype(:some_symbol) == Any
     @test prectype(1.0) == Float64
     @test prectype(big(1.0)) == BigFloat
     @test prectype(1) == typeof(float(1))
@@ -44,10 +66,26 @@ function test_prectype()
     @test promote_prectype(2) == 2
     @test promote_prectype(2, 3.0) isa Tuple{Float64,Float64}
     @test promote_prectype(2, 3.0+im, big(4)) isa Tuple{BigFloat,Complex{BigFloat},BigFloat}
+
+    @test to_prectype(Float64, Int) === Float64
+    @test to_prectype(Float32, Float64) === Float32
+    @test to_prectype(Int, Int) === Int
+    
+    @test to_prectype(Float64, Complex{Int}) === Complex{Float64}
+    @test to_prectype(Float32, Complex{Float64}) === Complex{Float32}
+    @test to_prectype(Float64, Vector{Int}) === Vector{Float64}
+    @test to_prectype(Float32, Vector{Float64}) === Vector{Float32}
+    @test to_prectype(Float64, SVector{3,Int}) === SVector{3,Float64}
+    @test to_prectype(Float32, MVector{3,Float64}) === MVector{3,Float32}
+    @test to_prectype(Float64, SMatrix{2,2,Int}) === SMatrix{2,2,Float64}
+    @test to_prectype(Float32, MMatrix{2,2,Float64}) === MMatrix{2,2,Float32}
+    @test_throws ArgumentError to_prectype(Float64, String)
+    @test_throws ArgumentError to_prectype(Float64, Dict{Int,Float64})    
 end
 
 
 function test_numtype()
+    @test numtype(:some_symbol) == Any
     @test numtype(1.0) == Float64
     @test numtype(big(1.0)) == BigFloat
     @test numtype(1) == Int
@@ -64,6 +102,7 @@ function test_numtype()
     @test numtype((1.0, 2.0, 3.0, 4.0)) == Float64
     @test numtype(1.0, big(2.0), 3.0+im) == Complex{BigFloat}
     @test numtype(typeof((1.0, big(2.0), 3.0+im))) == Complex{BigFloat}
+    @test numtype(Tuple{Int,Int,Int,Int,Float64}) == Float64
     @test @inferred(numtype(1, 2.0)) == Float64
     @test @inferred(numtype(typeof((1, 2.0, 3, 40+im)))) == Complex{Float64}
 
@@ -82,9 +121,34 @@ function test_numtype()
     @test promote_numtype(2) == 2
     @test promote_numtype(2, 3.0) isa Tuple{Float64,Float64}
     @test promote_numtype(2, 3.0+im, big(4)) isa Tuple{Complex{BigFloat},Complex{BigFloat},Complex{BigFloat}}
+
+    @test to_numtype(Float64, Int) === Float64
+    @test to_numtype(Float32, Float64) === Float32
+    @test to_numtype(Int, Int) === Int
+    
+    @test to_numtype(Float64, Vector{Int}) === Vector{Float64}
+    @test to_numtype(Float32, Vector{Float64}) === Vector{Float32}
+    @test to_numtype(Float64, SVector{3,Int}) === SVector{3,Float64}
+    @test to_numtype(Float32, MVector{3,Float64}) === MVector{3,Float32}
+    @test to_numtype(Float64, SMatrix{2,2,Int}) === SMatrix{2,2,Float64}
+    @test to_numtype(Float32, MMatrix{2,2,Float64}) === MMatrix{2,2,Float32}
+    @test_throws ArgumentError to_numtype(Float64, String)
+    @test_throws ArgumentError to_numtype(Float64, Dict{Int,Float64})
 end
 
-using FunctionMaps: isrealtype
+function test_eltype()
+    @test convert_eltype(Float64, Diagonal([1,2])) == Diagonal([1,2])
+    @test eltype(convert_eltype(Float64, Diagonal([1,2]))) == Float64
+    @test convert_eltype(Float64, 1:4) == 1:4
+    @test eltype(convert_eltype(Float64, 1:4)) == Float64
+    @test convert_eltype(Float64, Set([1,4])) == Set([1,4])
+    @test eltype(convert_eltype(Float64, Set([1,4]))) == Float64
+    @test convert_eltype(Float64, 5) == 5
+    @test eltype(convert_eltype(Float64, 5)) == Float64
+
+    @test FunctionMaps.promotable_eltypes(Int,Float64)
+    @test FunctionMaps.promotable_eltypes(Vector{Int},Vector{Float64})
+end
 
 function test_realtype()
     @test isrealtype(Any) == false
@@ -94,15 +158,10 @@ function test_realtype()
 end
 
 @testset "common functionality" begin
-    @testset "dimension" begin
-        test_dimension()
-    end
-    @testset "prectype" begin
-        test_prectype()
-    end
-    @testset "numtype" begin
-        test_numtype()
-    end
-
+    test_hashrec()
+    test_dimension()
+    test_prectype()
+    test_numtype()
+    test_eltype()
     test_realtype()
 end

test/test_generic.jl

@@ -1,3 +1,8 @@
+using FunctionMaps:
+    convert_domaintype, convert_codomaintype,
+    map_hash,
+    LazyInverse, jacobian!
+
 function generic_map_tests(T)
     for map in maps_to_test(T)
         @test prectype(map) == T