added trigonometric functions for poles

jishnub · Mar 4, 2020 · b61da52 · b61da52
1 parent f6a3abc
commit b61da52
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/ylm.jl b/src/ylm.jl
@@ -9,6 +9,12 @@ abstract type Pole <: Real end
 struct NorthPole <: Pole end
 struct SouthPole <: Pole end
 
+Base.cos(::NorthPole) = one(Float64)
+Base.cos(::SouthPole) = -one(Float64)
+Base.sin(::Pole) = zero(Float64)
+
+Base.one(::Type{<:Pole}) = one(Float64)
+
 const ContiguousArrayLike{T,N} = 
 	Union{Array{T,N}, Base.FastContiguousSubArray{T,N,<:Array{T}}}
 

diff --git a/test/test_ylm.jl b/test/test_ylm.jl
@@ -35,6 +35,10 @@ println()
 end # @testset
 
 @testset "Pole" begin
+   @testset "one" begin
+      @test one(NorthPole()) == 1       
+      @test one(SouthPole()) == 1       
+   end
    @testset "North pole" begin
       @testset "Plm" begin
          @test compute_p(10,1) ≈ compute_p(10,NorthPole())
@@ -43,6 +47,12 @@ end # @testset
          @test compute_y(10,1,0) ≈ compute_y(10,NorthPole())
          @test compute_y(10,1,0) ≈ compute_y(10,NorthPole(),π/2)
       end
+      @testset "trignometric functions" begin
+          @test cos(NorthPole()) == 1
+          @test sec(NorthPole()) == 1
+          @test sin(NorthPole()) == 0
+      end
+
    end
    @testset "south pole" begin
       @testset "Plm" begin
@@ -52,5 +62,10 @@ end # @testset
          @test compute_y(10,-1,0) ≈ compute_y(10,SouthPole())
          @test compute_y(10,-1,0) ≈ compute_y(10,SouthPole(),π/2)
       end
+      @testset "trignometric functions" begin
+          @test cos(SouthPole()) == -1
+          @test sec(SouthPole()) == -1
+          @test sin(SouthPole()) == 0
+      end
    end
 end