Skip to content
Prev Previous commit
Next Next commit
to specify the test result type to long
  • Loading branch information
gounnaa committed Sep 14, 2025
commit 8532f9b56b2e5eef6df645fd2add080f9205d2dc
15 changes: 9 additions & 6 deletions core/src/test/scala/org/graphframes/PatternMatchSuite.scala
Original file line number Diff line number Diff line change
Expand Up @@ -554,7 +554,10 @@ class PatternMatchSuite extends SparkFunSuite with GraphFrameTestSparkContext {
.select("u.id", "_v1.id", "_v2.id", "v.id")

val res = varEdge.collect().toSet
val expected = Set(Row(0, 1, 2, 0), Row(0, 1, 2, 3), Row(0, 1, 0, 1))
val expected = Set(
Row(0L, 1L, 2L, 0L),
Row(0L, 1L, 2L, 3L),
Row(0L, 1L, 0L, 1L))
compareResultToExpected(res, expected)
}

Expand All @@ -566,11 +569,11 @@ class PatternMatchSuite extends SparkFunSuite with GraphFrameTestSparkContext {

val res = varEdge.collect().toSet
val expected = Set(
Row(0, 1, 2, 0, 1, 0),
Row(0, 1, 0, 1, 0, 1),
Row(0, 1, 2, 0, 1, 2),
Row(0, 1, 0, 1, 2, 0),
Row(0, 1, 0, 1, 2, 3))
Row(0L, 1L, 2L, 0L, 1L, 0L),
Row(0L, 1L, 0L, 1L, 0L, 1L),
Row(0L, 1L, 2L, 0L, 1L, 2L),
Row(0L, 1L, 0L, 1L, 2L, 0L),
Row(0L, 1L, 0L, 1L, 2L, 3L))
compareResultToExpected(res, expected)
}

Expand Down
Loading