@@ -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
286286class HasInducedError (Params ):
0 commit comments