You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@spot.apache.org by ra...@apache.org on 2017/06/23 16:40:11 UTC
[41/50] [abbrv] incubator-spot git commit: SPOT-160 [ML] Upgrade
spot-ml Spark 2.1
http://git-wip-us.apache.org/repos/asf/incubator-spot/blob/e620c323/spot-ml/src/test/scala/org/apache/spot/utilities/FloatingPointUtility64.scala
----------------------------------------------------------------------
diff --git a/spot-ml/src/test/scala/org/apache/spot/utilities/FloatingPointUtility64.scala b/spot-ml/src/test/scala/org/apache/spot/utilities/FloatingPointUtility64.scala
index 2006fde..478129a 100644
--- a/spot-ml/src/test/scala/org/apache/spot/utilities/FloatingPointUtility64.scala
+++ b/spot-ml/src/test/scala/org/apache/spot/utilities/FloatingPointUtility64.scala
@@ -32,7 +32,7 @@ class FloatingPointUtility64 extends TestingSparkContextFlatSpec with Matchers {
"castColumn" should "return the exact same data frame" in {
- val testDataFrame = sqlContext.createDataFrame(Seq(("doc1", Array(1d, 2d)), ("doc2", Array(2d, 3d))))
+ val testDataFrame = sparkSession.createDataFrame(Seq(("doc1", Array(1d, 2d)), ("doc2", Array(2d, 3d))))
.withColumnRenamed("_1", DocumentName).withColumnRenamed("_2", TopicProbabilityMix)
val result = FloatPointPrecisionUtility64.castColumn(testDataFrame, TopicProbabilityMix)