Move TestUtils to a submodule

Author: gaurav-arya

Project.toml

@@ -5,6 +5,8 @@ version = "1.2.1"
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d"
 
 [compat]
 ChainRulesCore = "1"

src/AbstractFFTs.jl

@@ -9,5 +9,6 @@ export fft, ifft, bfft, fft!, ifft!, bfft!,
 
 include("definitions.jl")
 include("chainrules.jl")
+include("TestUtils.jl")
 
 end # module

test/AbstractFFTsTestUtils.jl renamed to src/TestUtils.jl

@@ -1,12 +1,11 @@
 # This file contains code that was formerly part of Julia. License is MIT: https://julialang.org/license
 
-module AbstractFFTsTestUtils
+module TestUtils
 
 export test_fft_backend
 
 using AbstractFFTs
 using AbstractFFTs: Plan
-using ChainRulesTestUtils
 
 using LinearAlgebra
 using Test
@@ -212,54 +211,6 @@ function test_fft_backend()
         f9(p::Plan{T}, sz) where {T} = AbstractFFTs.normalization(real(T), sz, fftdims(p))
         @test @inferred(f9(plan_fft(zeros(10), 1), 10)) == 1/10
     end
-
-    @testset "ChainRules" begin
-        @testset "shift functions" begin
-            for x in (randn(3), randn(3, 4), randn(3, 4, 5))
-                for dims in ((), 1, 2, (1,2), 1:2)
-                    any(d > ndims(x) for d in dims) && continue
-
-                    # type inference checks of `rrule` fail on old Julia versions
-                    # for higher-dimensional arrays:
-                    # https://github.com/JuliaMath/AbstractFFTs.jl/pull/58#issuecomment-916530016
-                    check_inferred = ndims(x) < 3 || VERSION >= v"1.6"
-
-                    test_frule(AbstractFFTs.fftshift, x, dims)
-                    test_rrule(AbstractFFTs.fftshift, x, dims; check_inferred=check_inferred)
-
-                    test_frule(AbstractFFTs.ifftshift, x, dims)
-                    test_rrule(AbstractFFTs.ifftshift, x, dims; check_inferred=check_inferred)
-                end
-            end
-        end
-
-        @testset "fft" begin
-            for x in (randn(3), randn(3, 4), randn(3, 4, 5))
-                N = ndims(x)
-                complex_x = complex.(x)
-                for dims in unique((1, 1:N, N))
-                    for f in (fft, ifft, bfft)
-                        test_frule(f, x, dims)
-                        test_rrule(f, x, dims)
-                        test_frule(f, complex_x, dims)
-                        test_rrule(f, complex_x, dims)
-                    end
-
-                    test_frule(rfft, x, dims)
-                    test_rrule(rfft, x, dims)
-
-                    for f in (irfft, brfft)
-                        for d in (2 * size(x, first(dims)) - 1, 2 * size(x, first(dims)) - 2)
-                            test_frule(f, x, d, dims)
-                            test_rrule(f, x, d, dims)
-                            test_frule(f, complex_x, d, dims)
-                            test_rrule(f, complex_x, d, dims)
-                        end
-                    end
-                end
-            end
-        end
-    end
 end
 
 end

test/chainrules.jl

@@ -0,0 +1,48 @@
+@testset "ChainRules" begin
+    @testset "shift functions" begin
+        for x in (randn(3), randn(3, 4), randn(3, 4, 5))
+            for dims in ((), 1, 2, (1,2), 1:2)
+                any(d > ndims(x) for d in dims) && continue
+
+                # type inference checks of `rrule` fail on old Julia versions
+                # for higher-dimensional arrays:
+                # https://github.com/JuliaMath/AbstractFFTs.jl/pull/58#issuecomment-916530016
+                check_inferred = ndims(x) < 3 || VERSION >= v"1.6"
+
+                test_frule(AbstractFFTs.fftshift, x, dims)
+                test_rrule(AbstractFFTs.fftshift, x, dims; check_inferred=check_inferred)
+
+                test_frule(AbstractFFTs.ifftshift, x, dims)
+                test_rrule(AbstractFFTs.ifftshift, x, dims; check_inferred=check_inferred)
+            end
+        end
+    end
+
+    @testset "fft" begin
+        for x in (randn(3), randn(3, 4), randn(3, 4, 5))
+            N = ndims(x)
+            complex_x = complex.(x)
+            for dims in unique((1, 1:N, N))
+                for f in (fft, ifft, bfft)
+                    test_frule(f, x, dims)
+                    test_rrule(f, x, dims)
+                    test_frule(f, complex_x, dims)
+                    test_rrule(f, complex_x, dims)
+                end
+
+                test_frule(rfft, x, dims)
+                test_rrule(rfft, x, dims)
+
+                for f in (irfft, brfft)
+                    for d in (2 * size(x, first(dims)) - 1, 2 * size(x, first(dims)) - 2)
+                        test_frule(f, x, d, dims)
+                        test_rrule(f, x, d, dims)
+                        test_frule(f, complex_x, d, dims)
+                        test_rrule(f, complex_x, d, dims)
+                    end
+                end
+            end
+        end
+    end
+end
+            

test/runtests.jl

@@ -1,12 +1,14 @@
 using Random
+using Test
+using AbstractFFTs
+using AbstractFFTs.TestUtils
+using ChainRulesTestUtils
 
 Random.seed!(1234)
 
 include("TestPlans.jl")
-include("AbstractFFTsTestUtils.jl")
 
 using .TestPlans
-using .AbstractFFTsTestUtils
-
 test_fft_backend()
 
+include("chainrules.jl")