add test for simplerasterdigraph

Author: Vincent Landau

src/rastergraphs.jl

@@ -162,15 +162,15 @@ increased distance between them (as compared to the distance between cardinal
 neighbors) is accounted for.
 """
 function simplerastergraph(
-    raster::GeoArray;
-    condition::Function = is_data,
-    directed::Bool = true,
-    cardinal_neighbors_only::Bool = false,
-)
-    v = make_vertex_raster(weight_raster)
+        raster::GeoArray;
+        condition::Function = is_data,
+        directed::Bool = true,
+        cardinal_neighbors_only::Bool = false,
+    )
+    v = make_vertex_raster(raster)
     g = make_simple_raster_graph(
         raster,
-        vertex_raster,
+        v,
         directed = directed,
         condition = condition,
         cardinal_neighbors_only = cardinal_neighbors_only

test/runtests.jl

@@ -1,9 +1,13 @@
 using Test, SpatialGraphs
 
-@testset "Raster Graph Construction" begin
-    include("rastergraphs.jl")
+@testset "Weighted Raster Graph Construction" begin
+    include("weightedrastergraphs.jl")
 end
 
-@testset "Raster DiGraph Construction" begin
-    include("rasterdigraphs.jl")
+@testset "Simple Raster DiGraph Construction" begin
+    include("simplerasterdigraphs.jl")
+end
+
+@testset "Weighted Raster DiGraph Construction" begin
+    include("weightedrasterdigraphs.jl")
 end

test/simplerasterdigraphs.jl

@@ -0,0 +1,54 @@
+using GeoData, LightGraphs, SimpleWeightedGraphs, SpatialGraphs, Test
+
+condition_array = Array{Float64}(undef, (3, 4, 1))
+condition_array[:,:,:] = [1, 3, 5, 2, 4, 8, 5, -9999, 2, 3, 6, 7]
+
+x = X(1:4)
+y = Y(1:3)
+band = Band(1:1)
+
+condition_raster = GeoArray(condition_array, (y, x, band), missingval = -9999)
+
+compare = <
+rasgraph = simplerastergraph(
+    condition_raster,
+    directed = true,
+    condition = compare
+)
+
+# no vertices in NoData pixels?
+@test (rasgraph.vertex_raster .== 0) == 
+        ((condition_raster .== condition_raster.missingval) .| 
+         isnan.(condition_raster))
+
+# Is the number of of vertices correct, and is the range of values correct?
+@test sort(collect(rasgraph.vertex_raster[rasgraph.vertex_raster .!= 0])) == 
+        collect(1:sum(
+            (condition_raster .!= condition_raster.missingval) .& 
+                (!).(isnan.(condition_raster))
+        ))
+
+graph_edges = collect(edges(rasgraph))
+
+# Test that the edges are correct and have proper weights
+for i in 1:length(graph_edges)
+    source_i = src(graph_edges[i])
+    dest_i = dst(graph_edges[i])
+
+    source_coords = findall(rasgraph.vertex_raster .== source_i)[1]
+    dest_coords = findall(rasgraph.vertex_raster .== dest_i)[1]
+
+    # Check that condition is met
+    @test compare(
+        condition_raster[source_coords],
+        condition_raster[dest_coords]
+    )
+
+    # Test that source row is within 1 step of dest row
+    row_diff = abs(source_coords[1] - dest_coords[1])
+    @test row_diff <= 1
+
+    # Test that source column is within 1 step of dest row
+    col_diff = abs(source_coords[2] - dest_coords[2])
+    @test col_diff <= 1
+end