Add Tests

Author: avik-pal

Project.toml

@@ -1,6 +1,6 @@
 name = "StaticArrays"
 uuid = "90137ffa-7385-5640-81b9-e52037218182"
-version = "1.7.1"
+version = "1.8.0"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -28,11 +28,12 @@ julia = "1.6"
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+ChainRulesTestUtils = "cdddcdb0-9152-4a09-a978-84456f9df70a"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d"
 
 [targets]
-test = ["InteractiveUtils", "Test", "BenchmarkTools", "OffsetArrays", "Statistics", "Unitful", "Aqua"]
+test = ["InteractiveUtils", "Test", "BenchmarkTools", "OffsetArrays", "Statistics", "Unitful", "Aqua", "ChainRulesTestUtils", "ChainRulesCore"]

ext/StaticArraysChainRulesCoreExt.jl

@@ -2,10 +2,10 @@ module StaticArraysChainRulesCoreExt
 
 using StaticArrays
 # ChainRulesCore imports
-import ChainRulesCore: ProjectTo, Tangent, project_type, rrule
+import ChainRulesCore: NoTangent, ProjectTo, Tangent, project_type, rrule
 import ChainRulesCore as CRC
 
-# Projecting a tuple to SMatrix leads to ChainRulesCore._projection_mismatch by default, so
+# Projecting a tuple to SMatrix leads to CRC._projection_mismatch by default, so
 # overloaded here
 function (project::ProjectTo{<:Tangent{<:Tuple}})(dx::StaticArraysCore.SArray)
     dy = reshape(dx, axes(project.elements))

src/StaticArrays.jl

@@ -133,8 +133,6 @@ include("flatten.jl")
 include("io.jl")
 include("pinv.jl")
 
-include("chainrules.jl")
-
 @static if !isdefined(Base, :get_extension) # VERSION < v"1.9-"
     include("../ext/StaticArraysStatisticsExt.jl")
 end

test/chainrules.jl

@@ -0,0 +1,10 @@
+using StaticArrays, ChainRulesCore, ChainRulesTestUtils, Test
+
+@testset "Chain Rules Integration" begin
+    @testset "Projection" begin
+        test_rrule(SMatrix{1, 4}, (1.0, 1.0, 1.0, 1.0))
+        test_rrule(SMatrix{4, 1}, (1.0, 1.0, 1.0, 1.0))
+        test_rrule(SMatrix{2, 2}, (1.0, 1.0, 1.0, 1.0))
+        test_rrule(SVector{4}, (1.0, 1.0, 1.0, 1.0))
+    end
+end

test/runtests.jl

@@ -88,4 +88,9 @@ if TEST_GROUP ∈ ["", "all", "group-B"]
     addtests("io.jl")
     addtests("svd.jl")
     addtests("unitful.jl")
+
+    # chain rules integration via pkg extensions is available only in Julia 1.9+
+    if VERSION ≥ v"1.9-"
+        addtests("chainrules.jl")
+    end
 end