Split into more testsets

Author: nickrobinson251

test/helper_functions.jl

@@ -1,33 +1,39 @@
 @testset "helper functions" begin
-    # Hits fallback, since we can't update `Diagonal`s in place
-    X = Diagonal([1, 1])
-    Y = copy(X)
-    @test ChainRules._update!(X, [1 2; 3 4]) == [2 2; 3 5]
-    @test X == Y  # no change to X
+    @testset "_update! Array" begin
+        # Hits fallback, since we can't update `Diagonal`s in place
+        X = Diagonal([1, 1])
+        Y = copy(X)
+        @test ChainRules._update!(X, [1 2; 3 4]) == [2 2; 3 5]
+        @test X == Y  # no change to X
 
-    X = [1 2; 3 4]
-    Y = copy(X)
-    @test ChainRules._update!(X, Diagonal([1, 1])) == [2 2; 3 5]
-    @test X != Y  # X has been updated
+        X = [1 2; 3 4]
+        Y = copy(X)
+        @test ChainRules._update!(X, Diagonal([1, 1])) == [2 2; 3 5]
+        @test X != Y  # X has been updated
+    end
 
-    # Reusing above X
-    @test ChainRules._update!(X, Zero()) === X
-    @test ChainRules._update!(Zero(), X) === X
-    @test ChainRules._update!(Zero(), Zero()) === Zero()
+    @testset "_update! Zero" begin
+        X = [1 2; 3 4]
+        @test ChainRules._update!(X, Zero()) === X
+        @test ChainRules._update!(Zero(), X) === X
+        @test ChainRules._update!(Zero(), Zero()) === Zero()
 
-    X = (A=[1 0; 0 1], B=[2 2; 2 2])
-    Y = deepcopy(X)
-    @test ChainRules._update!(X, Y) == (A=[2 0; 0 2], B=[4 4; 4 4])
-    @test X.A != Y.A
-    @test X.B != Y.B
+        X = (A=[1 0; 0 1], B=[2 2; 2 2])
+        Y = deepcopy(X)
+        @test ChainRules._update!(X, Y) == (A=[2 0; 0 2], B=[4 4; 4 4])
+        @test X.A != Y.A
+        @test X.B != Y.B
+    end
 
-    try
-        @eval cool(x,y)=x+y
-        # We defined a 2-arg method for `cool` but no `rrule`
-        ChainRules._checked_rrule(cool, 1.0, 2.0)
-    catch e
-        @test e isa ArgumentError
-        @test e.msg == "can't differentiate `cool(::Float64, ::Float64)`; no " *
-                       "matching `rrule` is defined"
+    @testset "_checked_rrule" begin
+        try
+            @eval cool(x,y)= x + y
+            # We defined a 2-arg method for `cool` but no `rrule`
+            ChainRules._checked_rrule(cool, 1.0, 2.0)
+        catch e
+            @test e isa ArgumentError
+            @test e.msg == "can't differentiate `cool(::Float64, ::Float64)`; no " *
+                "matching `rrule` is defined"
+        end
     end
 end