add tests

Author: Miha Zgubic

test/rulesets/Base/sort.jl

@@ -0,0 +1,15 @@
+@testset "sort.jl" begin
+    @testset "sort" begin
+        a = rand(10)
+        test_rrule(sort, a)
+        test_rrule(sort, a; fkwargs=(;rev=true))
+    end
+    @testset "partialsort" begin
+        a = rand(10)
+        test_rrule(partialsort, a, 4 ⊢ nothing)
+        test_rrule(partialsort, a, 3:5 ⊢ nothing)
+        test_rrule(partialsort, a, 1:2:6 ⊢ nothing)
+
+        test_rrule(partialsort, a, 4 ⊢ nothing, fkwargs=(;rev=true))
+    end
+end

test/runtests.jl

@@ -32,6 +32,7 @@ println("Testing ChainRules.jl")
             include_test("rulesets/Base/arraymath.jl")
             include_test("rulesets/Base/indexing.jl")
             include_test("rulesets/Base/mapreduce.jl")
+            include_test("rulesets/Base/sort.jl")
         end
         println()