Skip to content

Commit e713b24

Browse files
committed
Tweaked parquet I/O tests to trigger UDT issue.
1 parent 60c4917 commit e713b24

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,9 @@ trait TestEnvironment extends FunSpec with GeoTrellisTestEnvironment
6262
val dest = Files.createTempFile(Paths.get(outputLocalPath), "rf", ".parquet")
6363
logger.trace(s"Writing '${sanitized.columns.mkString(", ")}' to '$dest'...")
6464
sanitized.write.mode(SaveMode.Overwrite).parquet(dest.toString)
65-
val rows = df.sparkSession.read.parquet(dest.toString).count()
65+
val rows = df.sparkSession.read.parquet(dest.toString).collect()
6666
logger.trace(s" read back $rows row(s)")
67-
rows == inRows
67+
rows.length == inRows
6868
}
6969

7070
/**

0 commit comments

Comments
 (0)