add mod2pi and its test

JuliaDiff · May 9, 2019 · 8b6b48e · 8b6b48e
1 parent a9b00c8
commit 8b6b48e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/rules.jl b/src/rules.jl
@@ -58,6 +58,7 @@
 @define_diffrule Base.acsch(x)                = :( -inv(abs($x) * sqrt(1 + $x^2))      )
 @define_diffrule Base.acoth(x)                = :(  inv(1 - $x^2)                      )
 @define_diffrule Base.deg2rad(x)              = :(  π / 180                            )
+@define_diffrule Base.mod2pi(x)               = :(  isinteger($x / 2pi) ? NaN : 1      )
 @define_diffrule Base.rad2deg(x)              = :(  180 / π                            )
 if VERSION < v"0.7-"
     @define_diffrule Base.gamma(x)            = :(  digamma($x) * gamma($x)            )

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -27,6 +27,11 @@ for (M, f, arity) in DiffRules.diffrules()
         @eval begin
             goo = rand() + $modifier
             @test isapprox($deriv, finitediff($M.$f, goo), rtol=0.05)
+            # test for 2pi functions
+            if occursin("mod2pi", string($M.$f))
+                goo = 4pi + $modifier
+                @test NaN === $deriv
+            end
         end
     elseif arity == 2
         @test DiffRules.hasdiffrule(M, f, 2)