view test/Grids/manifolds_test.jl @ 1781:a73838c9ef94 feature/grids/manifolds

Let Chart implement Base.ndims instead of domain_dim
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 16 Sep 2024 09:05:47 +0200
parents 8ecdc5bb46be
children ea98f03e18e0
line wrap: on
line source

using Test

using Diffinitive.Grids
using Diffinitive.RegionIndices
using Diffinitive.LazyTensors

# using StaticArrays

@testset "ParameterSpace" begin
    @test ndims(HyperBox([1,1], [2,2])) == 2
    @test ndims(unittetrahedron()) == 3
end

@testset "Interval" begin
    @test Interval <: ParameterSpace{1}

    @test Interval(0,1) isa Interval{Int}
    @test Interval(0,1.) isa Interval{Float64}

    @test unitinterval() isa Interval{Float64}
    @test unitinterval() == Interval(0.,1.)
    @test limits(unitinterval()) == (0.,1.)

    @test unitinterval(Int) isa Interval{Int}
    @test unitinterval(Int) == Interval(0,1)
    @test limits(unitinterval(Int)) == (0,1)
end

@testset "HyperBox" begin
    @test HyperBox{<:Any, 2} <: ParameterSpace{2}
    @test HyperBox([1,1], [2,2]) isa HyperBox{Int, 2}

    @test HyperBox([1,2], [1.,2.]) isa HyperBox{Float64,2}

    @test limits(HyperBox([1,2], [3,4])) == ([1,2], [3,4])
    @test limits(HyperBox([1,2], [3,4]), 1) == (1,3)
    @test limits(HyperBox([1,2], [3,4]), 2) == (2,4)

    @test unitsquare() isa HyperBox{Float64,2}
    @test limits(unitsquare()) == ([0,0],[1,1])

    @test unitcube() isa HyperBox{Float64,3}
    @test limits(unitcube()) == ([0,0,0],[1,1,1])

    @test unithyperbox(4) isa HyperBox{Float64,4}
    @test limits(unithyperbox(4)) == ([0,0,0,0],[1,1,1,1])
end

@testset "Simplex" begin
    @test Simplex{<:Any, 3} <: ParameterSpace{3}
    @test Simplex([1,2], [3,4]) isa Simplex{Int, 2}
    @test Simplex([1,2,3], [4,5,6],[1,1,1]) isa Simplex{Int, 3}

    @test Simplex([1,2], [3.,4.]) isa Simplex{Float64, 2}

    @test verticies(Simplex([1,2], [3,4])) == ([1,2], [3,4])

    @test unittriangle() isa Simplex{Float64,2}
    @test verticies(unittriangle()) == ([0,0], [1,0], [0,1])

    @test unittetrahedron() isa  Simplex{Float64,3}
    @test verticies(unittetrahedron()) == ([0,0,0], [1,0,0], [0,1,0],[0,0,1])

    @test unitsimplex(4) isa Simplex{Float64,4}
end

@testset "Chart" begin
    c = Chart(x->2x, unitsquare())
    @test c isa Chart{2}
    @test c([3,2]) == [6,4]
    @test parameterspace(c) == unitsquare()
    @test ndims(c) == 2
end

@testset "Atlas" begin

end