add printing and tests for all models

Author: mileslucas

Project.toml

@@ -8,6 +8,7 @@ ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 CoordinateTransformations = "150eb455-5306-5404-9cee-2592286d6298"
 Distances = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7"
 KeywordCalls = "4d827475-d3e4-43d6-abe3-9688362ede9f"
+LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 RecipesBase = "3cdcf5f2-1ef4-517c-9805-6587b60abb01"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"

src/airy.jl

@@ -43,6 +43,9 @@ end
 Base.size(a::AiryDisk) = map(length, a.indices)
 Base.axes(a::AiryDisk) = a.indices
 
+# short printing
+Base.show(io::IO, a::AiryDisk{T}) where {T} = print(io, "AiryDisk{$T}(pos=$(a.pos), fwhm=$(a.fwhm), amp=$(a.amp))")
+
 const rz = 3.8317059702075125 / π
 
 function (a::AiryDisk{T})(point::AbstractVector) where T

src/moffat.jl

@@ -39,6 +39,10 @@ end
 Base.size(m::Moffat) = map(length, m.indices)
 Base.axes(m::Moffat) = m.indices
 
+# short printing
+Base.show(io::IO, m::Moffat{T}) where {T} = print(io, "Moffat{$T}(pos=$(m.pos), fwhm=$(m.fwhm), amp=$(m.amp), alpha=$(m.alpha))")
+
+
 # scalar case
 function (m::Moffat{T})(point::AbstractVector) where T
     hwhm = m.fwhm / 2

test/runtests.jl

@@ -82,11 +82,13 @@ end
     m = Gaussian(fwhm=10)
     expected = exp(-4 * log(2) * sum(abs2, SA[1, 2]) / 100)
     @test m[2, 1] ≈ m(1, 2) ≈ expected
-
+    @test repr(m) == "Gaussian{Float64}(pos=[0, 0], fwhm=10, amp=1.0)"
+    
     m = Gaussian(fwhm=(10, 9))
     wdist = (1/10)^2 + (2/9)^2
     expected = exp(-4 * log(2) * wdist)
     @test m[2, 1] ≈ m(1, 2) ≈ expected
+    @test repr(m) == "Gaussian{Float64}(pos=[0, 0], fwhm=(10, 9), amp=1.0)"
 
     # test Normal alias
     @test Normal(fwhm=10) === Gaussian(fwhm=10)
@@ -101,6 +103,7 @@ end
     @test m(-radius, 0) ≈ 0 atol=eps(Float64)
     @test m(0, radius) ≈ 0 atol=eps(Float64)
     @test m(0, -radius) ≈ 0 atol=eps(Float64)
+    @test repr(m) == "AiryDisk{Float64}(pos=[0, 0], fwhm=10, amp=1.0)"
 
     m = AiryDisk(fwhm=(10, 9))
     r1 = m.fwhm[1] * 1.18677
@@ -110,22 +113,26 @@ end
     @test m(-r1, 0) ≈ 0 atol=eps(Float64)
     @test m(0, r2) ≈ 0 atol=eps(Float64)
     @test m(0, -r2) ≈ 0 atol=eps(Float64)
+    @test repr(m) == "AiryDisk{Float64}(pos=[0, 0], fwhm=(10, 9), amp=1.0)"
 end
 
 @testset "Moffat" begin
     m = Moffat(fwhm=10)
     expected = inv(1 + sum(abs2, SA[1, 2]) / 25)
     @test m[2, 1] ≈ m(1, 2) ≈ expected
+    @test repr(m) == "Moffat{Float64}(pos=[0, 0], fwhm=10, amp=1.0, alpha=1)"
 
     m = Moffat(fwhm=(10, 9))
     wdist = (1/5)^2 + (2/4.5)^2
     expected = inv(1 + wdist)
     @test m[2, 1] ≈ m(1, 2) ≈ expected
+    @test repr(m) == "Moffat{Float64}(pos=[0, 0], fwhm=(10, 9), amp=1.0, alpha=1)"
 
     # different alpha
     m = Moffat(fwhm=10, alpha=2)
     expected = inv(1 + sum(abs2, SA[1, 2]) / 25)^2
     @test m[2, 1] ≈ m(1, 2) ≈ expected
+    @test repr(m) == "Moffat{Float64}(pos=[0, 0], fwhm=10, amp=1.0, alpha=2)"
 end
 
 include("plotting.jl")