Merge pull request #55 from jishnub/showFreq

Compact show for Frequencies
JuliaMath · Feb 5, 2024 · d743eb6 · d743eb6
2 parents d9f15f4 + 41f58b5
commit d743eb6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/definitions.jl b/src/definitions.jl
@@ -462,6 +462,10 @@ Base.maximum(f::Frequencies{T}) where T = (f.n_nonnegative - ifelse(f.multiplier
 Base.minimum(f::Frequencies{T}) where T = (f.n_nonnegative - ifelse(f.multiplier >= zero(T), f.n, 1)) * f.multiplier
 Base.extrema(f::Frequencies) = (minimum(f), maximum(f))
 
+function show(io::IO, f::Frequencies)
+    print(io, Frequencies, "(", f.n_nonnegative, ", ", f.n, ", ", f.multiplier, ")")
+end
+
 """
     fftfreq(n, fs=1)
 Return the discrete Fourier transform (DFT) sample frequencies for a DFT of length `n`. The returned

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -108,6 +108,13 @@ end
             check_extrema(freqs)
         end
     end
+
+    @testset "show" begin
+        @test repr(fftfreq(6)) == "Frequencies(3, 6, $(1/6))"
+        @test repr(fftfreq(7, 2)) == "Frequencies(4, 7, $(2/7))"
+        @test repr(rfftfreq(5, 0.3)) == "Frequencies(3, 3, $(0.3/5))"
+        @test repr(rfftfreq(4, 3)) == "Frequencies(3, 3, $(3/4))"
+    end
 end
 
 @testset "normalization" begin