Change the place of testing code

avulanov · Jul 15, 2015 · 28bd165 · 28bd165
1 parent d7a332a
commit 28bd165
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/python/pyspark/mllib/clustering.py b/python/pyspark/mllib/clustering.py
@@ -565,17 +565,6 @@ def predictOnValues(self, dstream):
         return dstream.mapValues(lambda x: self._model.predict(x))
 
 
-def _test():
-    import doctest
-    import pyspark.mllib.clustering
-    globs = pyspark.mllib.clustering.__dict__.copy()
-    globs['sc'] = SparkContext('local[4]', 'PythonTest', batchSize=2)
-    (failure_count, test_count) = doctest.testmod(globs=globs, optionflags=doctest.ELLIPSIS)
-    globs['sc'].stop()
-    if failure_count:
-        exit(-1)
-
-
 class LDAModel(JavaModelWrapper):
 
     """ A clustering model derived from the LDA method.
@@ -638,5 +627,16 @@ def train(cls, rdd, k=10, maxIterations=20, docConcentration=-1.0,
         return LDAModel(model)
 
 
+def _test():
+    import doctest
+    import pyspark.mllib.clustering
+    globs = pyspark.mllib.clustering.__dict__.copy()
+    globs['sc'] = SparkContext('local[4]', 'PythonTest', batchSize=2)
+    (failure_count, test_count) = doctest.testmod(globs=globs, optionflags=doctest.ELLIPSIS)
+    globs['sc'].stop()
+    if failure_count:
+        exit(-1)
+
+
 if __name__ == "__main__":
     _test()