Fix bug in lenses and add simplistic tests

src/lenses.jl

@@ -30,8 +30,8 @@ function lens(f::Number, L::Number, size::Tuple=(512, 512); radius=Inf, λ=550e-
     out = zeros(ComplexF64, size)
 
     # iterate over values
-    for (j, x) in enumerate(fftpos(L, size(arr)[2]))
-        for (i, y) in enumerate(range(L, size(arr)[1]))
+    for (j, x) in enumerate(fftpos(L, size[2]))
+        for (i, y) in enumerate(fftpos(L, size[1]))
             out[i, j] = circ(radius, x, y) * lens(f, x, y; radius=radius, λ=λ, n=n)
         end
     end

test/lenses.jl

@@ -0,0 +1,11 @@
+@testset "Test ideal lens" begin
+
+    arr = randn((10, 10))
+    @test all(abs.(lens(10.0, 12.0, size(arr), radius=1000)) .≈ 1)
+
+    arr2 = zeros(size(arr))
+    arr2[6, 6] = 1
+    @test arr2 ≈ abs.(lens(10.0, 12.0, size(arr), radius=0))
+
+
+end

test/runtests.jl

@@ -7,14 +7,15 @@ using Random
 Random.seed!(42)
 
 tests = [
-    "convolutions",
-    "utils",
-    "psf",
     "apertures",
+    "convolutions",
+    "lenses",
     "physical_conversions",
-    "resolution_equations",
     "propagations",
-    "utils_view"
+    "psf",
+    "resolution_equations",
+    "utils",
+    "utils_view",
 ]
 
 for t in tests