aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/ml/tests.py
diff options
context:
space:
mode:
authorYanbo Liang <ybliang8@gmail.com>2015-09-18 09:53:52 -0700
committerXiangrui Meng <meng@databricks.com>2015-09-18 09:53:52 -0700
commit35e8ab939000d4a1a01c1af4015c25ff6f4013a3 (patch)
treeaf13816f2009349515257d5b6a2c38b39e1bb6a8 /python/pyspark/ml/tests.py
parent20fd35dfd1ac402b622604e7bbedcc53a580b0a2 (diff)
downloadspark-35e8ab939000d4a1a01c1af4015c25ff6f4013a3.tar.gz
spark-35e8ab939000d4a1a01c1af4015c25ff6f4013a3.tar.bz2
spark-35e8ab939000d4a1a01c1af4015c25ff6f4013a3.zip
[SPARK-10615] [PYSPARK] change assertEquals to assertEqual
As ```assertEquals``` is deprecated, so we need to change ```assertEquals``` to ```assertEqual``` for existing python unit tests. Author: Yanbo Liang <ybliang8@gmail.com> Closes #8814 from yanboliang/spark-10615.
Diffstat (limited to 'python/pyspark/ml/tests.py')
-rw-r--r--python/pyspark/ml/tests.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/python/pyspark/ml/tests.py b/python/pyspark/ml/tests.py
index b892318f50..648fa8858f 100644
--- a/python/pyspark/ml/tests.py
+++ b/python/pyspark/ml/tests.py
@@ -182,7 +182,7 @@ class ParamTests(PySparkTestCase):
self.assertEqual(testParams.getMaxIter(), 10)
testParams.setMaxIter(100)
self.assertTrue(testParams.isSet(maxIter))
- self.assertEquals(testParams.getMaxIter(), 100)
+ self.assertEqual(testParams.getMaxIter(), 100)
self.assertTrue(testParams.hasParam(inputCol))
self.assertFalse(testParams.hasDefault(inputCol))
@@ -195,7 +195,7 @@ class ParamTests(PySparkTestCase):
testParams._setDefault(seed=41)
testParams.setSeed(43)
- self.assertEquals(
+ self.assertEqual(
testParams.explainParams(),
"\n".join(["inputCol: input column name (undefined)",
"maxIter: max number of iterations (>= 0) (default: 10, current: 100)",
@@ -264,23 +264,23 @@ class FeatureTests(PySparkTestCase):
self.assertEqual(ngram0.getInputCol(), "input")
self.assertEqual(ngram0.getOutputCol(), "output")
transformedDF = ngram0.transform(dataset)
- self.assertEquals(transformedDF.head().output, ["a b c d", "b c d e"])
+ self.assertEqual(transformedDF.head().output, ["a b c d", "b c d e"])
def test_stopwordsremover(self):
sqlContext = SQLContext(self.sc)
dataset = sqlContext.createDataFrame([Row(input=["a", "panda"])])
stopWordRemover = StopWordsRemover(inputCol="input", outputCol="output")
# Default
- self.assertEquals(stopWordRemover.getInputCol(), "input")
+ self.assertEqual(stopWordRemover.getInputCol(), "input")
transformedDF = stopWordRemover.transform(dataset)
- self.assertEquals(transformedDF.head().output, ["panda"])
+ self.assertEqual(transformedDF.head().output, ["panda"])
# Custom
stopwords = ["panda"]
stopWordRemover.setStopWords(stopwords)
- self.assertEquals(stopWordRemover.getInputCol(), "input")
- self.assertEquals(stopWordRemover.getStopWords(), stopwords)
+ self.assertEqual(stopWordRemover.getInputCol(), "input")
+ self.assertEqual(stopWordRemover.getStopWords(), stopwords)
transformedDF = stopWordRemover.transform(dataset)
- self.assertEquals(transformedDF.head().output, ["a"])
+ self.assertEqual(transformedDF.head().output, ["a"])
class HasInducedError(Params):