Merge pull request #300 from s22s/feature/rf_explode_tiles_sample

Add unit test for rf_explode_tiles_resample

Author: metasim

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

@@ -44,6 +44,7 @@ class ExplodeSpec extends TestEnvironment with TestData {
           |""".stripMargin)
       write(query)
       assert(query.select("cell_0", "cell_1").as[(Double, Double)].collect().forall(_ == ((1.0, 2.0))))
+      query.select("cell_0", "cell_1").count() should be (100L)
       val query2 = sql(
         """|select rf_dimensions(tiles) as dims, rf_explode_tiles(tiles) from (
            |select rf_make_constant_tile(1, 10, 10, 'int8raw') as tiles)
@@ -66,6 +67,20 @@ class ExplodeSpec extends TestEnvironment with TestData {
       assert(exploded.count() < 9)
     }
 
+    ignore("should explode tiles with random sampling in SQL API") {
+      // was pretty much a WONT FIX from issue 97
+      val df = Seq[(Tile, Tile)]((byteArrayTile, byteArrayTile)).toDF("tile1", "tile2")
+      val exploded = df.selectExpr("rf_explode_tiles_sample(0.5, tile1, tile2)")
+      logger.info("rf_explode_tiles schema with double frac arg \n" + exploded.schema.treeString)
+      assert(exploded.columns.length === 4)
+      assert(exploded.count() < 9)
+
+      val explodedSeed = df.selectExpr("rf_explode_tiles_sample(0.5, 784505, tile1, tile2)")
+      assert(explodedSeed.columns.length === 4)
+      logger.info(s"Count with seed 784505: ${explodedSeed.count().toString}")
+      assert(explodedSeed.count() < 9)
+    }
+
     it("should handle null tiles") {
       val df = Seq[Tile](null, byteArrayTile, null, byteArrayTile, null).toDF("tile1")
       val exploded = df.select(rf_explode_tiles($"tile1"))