diff options
author | Xiangrui Meng <meng@databricks.com> | 2015-11-15 13:23:05 -0800 |
---|---|---|
committer | Yin Huai <yhuai@databricks.com> | 2015-11-15 13:23:05 -0800 |
commit | 64e55511033afb6ef42be142eb371bfbc31f5230 (patch) | |
tree | bf46ac14f25d749fa4b14c36cf4afa01b5be6ba7 /mllib/src/test | |
parent | d22fc10887fdc6a86f6122648a823d0d37d4d795 (diff) | |
download | spark-64e55511033afb6ef42be142eb371bfbc31f5230.tar.gz spark-64e55511033afb6ef42be142eb371bfbc31f5230.tar.bz2 spark-64e55511033afb6ef42be142eb371bfbc31f5230.zip |
[SPARK-11672][ML] set active SQLContext in JavaDefaultReadWriteSuite
The same as #9694, but for Java test suite. yhuai
Author: Xiangrui Meng <meng@databricks.com>
Closes #9719 from mengxr/SPARK-11672.4.
Diffstat (limited to 'mllib/src/test')
-rw-r--r-- | mllib/src/test/java/org/apache/spark/ml/util/JavaDefaultReadWriteSuite.java | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/mllib/src/test/java/org/apache/spark/ml/util/JavaDefaultReadWriteSuite.java b/mllib/src/test/java/org/apache/spark/ml/util/JavaDefaultReadWriteSuite.java index c39538014b..01ff1ea658 100644 --- a/mllib/src/test/java/org/apache/spark/ml/util/JavaDefaultReadWriteSuite.java +++ b/mllib/src/test/java/org/apache/spark/ml/util/JavaDefaultReadWriteSuite.java @@ -32,17 +32,23 @@ import org.apache.spark.util.Utils; public class JavaDefaultReadWriteSuite { JavaSparkContext jsc = null; + SQLContext sqlContext = null; File tempDir = null; @Before public void setUp() { jsc = new JavaSparkContext("local[2]", "JavaDefaultReadWriteSuite"); + SQLContext.clearActive(); + sqlContext = new SQLContext(jsc); + SQLContext.setActive(sqlContext); tempDir = Utils.createTempDir( System.getProperty("java.io.tmpdir"), "JavaDefaultReadWriteSuite"); } @After public void tearDown() { + sqlContext = null; + SQLContext.clearActive(); if (jsc != null) { jsc.stop(); jsc = null; @@ -64,7 +70,6 @@ public class JavaDefaultReadWriteSuite { } catch (IOException e) { // expected } - SQLContext sqlContext = new SQLContext(jsc); instance.write().context(sqlContext).overwrite().save(outputPath); MyParams newInstance = MyParams.load(outputPath); Assert.assertEquals("UID should match.", instance.uid(), newInstance.uid()); |