Skip to content

Commit 35e8ab9

Browse files
yanboliangmengxr
authored andcommitted
[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 <[email protected]> Closes #8814 from yanboliang/spark-10615.
1 parent 20fd35d commit 35e8ab9

File tree

4 files changed

+99
-99
lines changed

4 files changed

+99
-99
lines changed

python/pyspark/ml/tests.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def test_params(self):
182182
self.assertEqual(testParams.getMaxIter(), 10)
183183
testParams.setMaxIter(100)
184184
self.assertTrue(testParams.isSet(maxIter))
185-
self.assertEquals(testParams.getMaxIter(), 100)
185+
self.assertEqual(testParams.getMaxIter(), 100)
186186

187187
self.assertTrue(testParams.hasParam(inputCol))
188188
self.assertFalse(testParams.hasDefault(inputCol))
@@ -195,7 +195,7 @@ def test_params(self):
195195
testParams._setDefault(seed=41)
196196
testParams.setSeed(43)
197197

198-
self.assertEquals(
198+
self.assertEqual(
199199
testParams.explainParams(),
200200
"\n".join(["inputCol: input column name (undefined)",
201201
"maxIter: max number of iterations (>= 0) (default: 10, current: 100)",
@@ -264,23 +264,23 @@ def test_ngram(self):
264264
self.assertEqual(ngram0.getInputCol(), "input")
265265
self.assertEqual(ngram0.getOutputCol(), "output")
266266
transformedDF = ngram0.transform(dataset)
267-
self.assertEquals(transformedDF.head().output, ["a b c d", "b c d e"])
267+
self.assertEqual(transformedDF.head().output, ["a b c d", "b c d e"])
268268

269269
def test_stopwordsremover(self):
270270
sqlContext = SQLContext(self.sc)
271271
dataset = sqlContext.createDataFrame([Row(input=["a", "panda"])])
272272
stopWordRemover = StopWordsRemover(inputCol="input", outputCol="output")
273273
# Default
274-
self.assertEquals(stopWordRemover.getInputCol(), "input")
274+
self.assertEqual(stopWordRemover.getInputCol(), "input")
275275
transformedDF = stopWordRemover.transform(dataset)
276-
self.assertEquals(transformedDF.head().output, ["panda"])
276+
self.assertEqual(transformedDF.head().output, ["panda"])
277277
# Custom
278278
stopwords = ["panda"]
279279
stopWordRemover.setStopWords(stopwords)
280-
self.assertEquals(stopWordRemover.getInputCol(), "input")
281-
self.assertEquals(stopWordRemover.getStopWords(), stopwords)
280+
self.assertEqual(stopWordRemover.getInputCol(), "input")
281+
self.assertEqual(stopWordRemover.getStopWords(), stopwords)
282282
transformedDF = stopWordRemover.transform(dataset)
283-
self.assertEquals(transformedDF.head().output, ["a"])
283+
self.assertEqual(transformedDF.head().output, ["a"])
284284

285285

286286
class HasInducedError(Params):

0 commit comments

Comments
 (0)