fix test: unwrap AxisArray

Author: johnnychen94

test/cuda/models.jl

@@ -35,7 +35,7 @@ using ImageFiltering.Models
         @test Array(img_smoothed_cu) ≈ img_smoothed
 
         # 3D Gray
-        img = restrict(testimage("mri"), (1, 2))
+        img = Gray.(restrict(testimage("mri"), (1, 2)))
         img_noisy = img .+ 0.05randn(MersenneTwister(0), size(img))
         img_smoothed = solve_ROF_PD(img_noisy, 0.02, 20)
         @test ndims(img_smoothed) == 3

test/models.jl

@@ -23,7 +23,7 @@ using ImageFiltering.Models
         @test assess_ssim(img_smoothed, img) > 0.90
 
         # 3D Gray
-        img = restrict(testimage("mri"), (1, 2))
+        img = Gray.(restrict(testimage("mri"), (1, 2)))
         img_noisy = img .+ 0.05randn(MersenneTwister(0), size(img))
         img_smoothed = solve_ROF_PD(img_noisy, 0.02, 20)
         @test ndims(img_smoothed) == 3