Update runtests

Author: roflmaostc

test/runtests.jl

@@ -118,14 +118,14 @@ using Plots
         @test ABCDMatrixOptics.zR(beam4) /2 ≈ ABCDMatrixOptics.zR(GaussianBeam())
     end
 
-    # how to do that properly?
+    # TODO: how to do that properly?
     @testset "Plots" begin
-        p = plot([FreeSpace(100e-3)], GaussianBeam(w0=100e-6, λ=500e-9, n=1.3, zpos=0))
-        @test p == p
-        p = plot([FreeSpace(100e-3)], GaussianBeam(w0=100e-6, λ=1000e-9, n=1.3, zpos=0))
-        @test p == p
-        p = plot([FreeSpace(100e-3)], GaussianBeam(w0=100e-6, λ=100e-9, n=1.3, zpos=0))
-        @test p == p
+        for λ  in [100e-9, 500e-9] 
+            for beam in [GaussianBeam(w0=100e-6, λ=λ, n=1.3, zpos=0), GeometricBeam(w=100e-6, k=0.0)]
+                p = plot([FreeSpace(100e-3)], beam) 
+                @test p == p
+            end
+        end
     end
 
     return true