Add failing unit test for issue 333, error in rf_agg_local_mean

Signed-off-by: Jason T. Brown <[email protected]>

Author: vpipkt

core/src/test/scala/org/locationtech/rasterframes/RasterFunctionsSpec.scala

@@ -525,6 +525,23 @@ class RasterFunctionsSpec extends TestEnvironment with RasterMatchers {
       checkDocs("rf_agg_local_max")
     }
 
+    ignore("should compute local mean") {
+      checkDocs("rf_agg_local_mean")
+      // https://github.com/locationtech/rasterframes/issues/333
+      val df = Seq(two, three, one, six).toDF("tile")
+          .withColumn("id", monotonically_increasing_id())
+      df.select(rf_agg_local_mean($"tile")).first() should be(three.toArrayTile())
+
+      df.selectExpr("rf_agg_local_mean(tile)").as[Tile].first() should be(three.toArrayTile())
+
+      noException should be thrownBy {
+        df.groupBy($"id")
+          .agg(rf_agg_local_mean($"tile"))
+          .collect()
+      }
+
+    }
+
     it("should compute local data cell counts") {
       val df = Seq(two, randNDPRT, nd).toDF("tile")
       val t1 = df.select(rf_agg_local_data_cells($"tile")).first()