aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/mllib/tests.py
diff options
context:
space:
mode:
authorReynold Xin <rxin@apache.org>2014-05-25 17:15:01 -0700
committerReynold Xin <rxin@apache.org>2014-05-25 17:15:01 -0700
commitd33d3c61ae9e4551aed0217e525a109e678298f2 (patch)
tree109ffeeaf31ae267bbe791051fd39f490af04aa4 /python/pyspark/mllib/tests.py
parent14f0358b2a0a9b92526bdad6d501ab753459eaa0 (diff)
downloadspark-d33d3c61ae9e4551aed0217e525a109e678298f2.tar.gz
spark-d33d3c61ae9e4551aed0217e525a109e678298f2.tar.bz2
spark-d33d3c61ae9e4551aed0217e525a109e678298f2.zip
Fix PEP8 violations in Python mllib.
Author: Reynold Xin <rxin@apache.org> Closes #871 from rxin/mllib-pep8 and squashes the following commits: 848416f [Reynold Xin] Fixed a typo in the previous cleanup (c -> sc). a8db4cd [Reynold Xin] Fix PEP8 violations in Python mllib.
Diffstat (limited to 'python/pyspark/mllib/tests.py')
-rw-r--r--python/pyspark/mllib/tests.py27
1 files changed, 10 insertions, 17 deletions
diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py
index 1ee96bb4af..37ccf1d590 100644
--- a/python/pyspark/mllib/tests.py
+++ b/python/pyspark/mllib/tests.py
@@ -23,7 +23,7 @@ from numpy import array, array_equal
import unittest
from pyspark.mllib._common import _convert_vector, _serialize_double_vector, \
- _deserialize_double_vector, _dot, _squared_distance
+ _deserialize_double_vector, _dot, _squared_distance
from pyspark.mllib.linalg import SparseVector
from pyspark.mllib.regression import LabeledPoint
from pyspark.tests import PySparkTestCase
@@ -46,12 +46,9 @@ class VectorTests(unittest.TestCase):
self.assertTrue(sv is _convert_vector(sv))
self.assertTrue(dv is _convert_vector(dv))
self.assertTrue(array_equal(dv, _convert_vector(lst)))
- self.assertEquals(sv,
- _deserialize_double_vector(_serialize_double_vector(sv)))
- self.assertTrue(array_equal(dv,
- _deserialize_double_vector(_serialize_double_vector(dv))))
- self.assertTrue(array_equal(dv,
- _deserialize_double_vector(_serialize_double_vector(lst))))
+ self.assertEquals(sv, _deserialize_double_vector(_serialize_double_vector(sv)))
+ self.assertTrue(array_equal(dv, _deserialize_double_vector(_serialize_double_vector(dv))))
+ self.assertTrue(array_equal(dv, _deserialize_double_vector(_serialize_double_vector(lst))))
def test_dot(self):
sv = SparseVector(4, {1: 1, 3: 2})
@@ -132,7 +129,7 @@ class ListTests(PySparkTestCase):
def test_regression(self):
from pyspark.mllib.regression import LinearRegressionWithSGD, LassoWithSGD, \
- RidgeRegressionWithSGD
+ RidgeRegressionWithSGD
data = [
LabeledPoint(-1.0, [0, -1]),
LabeledPoint(1.0, [0, 1]),
@@ -179,14 +176,10 @@ class SciPyTests(PySparkTestCase):
self.assertEquals(sv, _convert_vector(lil.tocoo()))
self.assertEquals(sv, _convert_vector(lil.tocsr()))
self.assertEquals(sv, _convert_vector(lil.todok()))
- self.assertEquals(sv,
- _deserialize_double_vector(_serialize_double_vector(lil)))
- self.assertEquals(sv,
- _deserialize_double_vector(_serialize_double_vector(lil.tocsc())))
- self.assertEquals(sv,
- _deserialize_double_vector(_serialize_double_vector(lil.tocsr())))
- self.assertEquals(sv,
- _deserialize_double_vector(_serialize_double_vector(lil.todok())))
+ self.assertEquals(sv, _deserialize_double_vector(_serialize_double_vector(lil)))
+ self.assertEquals(sv, _deserialize_double_vector(_serialize_double_vector(lil.tocsc())))
+ self.assertEquals(sv, _deserialize_double_vector(_serialize_double_vector(lil.tocsr())))
+ self.assertEquals(sv, _deserialize_double_vector(_serialize_double_vector(lil.todok())))
def test_dot(self):
from scipy.sparse import lil_matrix
@@ -265,7 +258,7 @@ class SciPyTests(PySparkTestCase):
def test_regression(self):
from pyspark.mllib.regression import LinearRegressionWithSGD, LassoWithSGD, \
- RidgeRegressionWithSGD
+ RidgeRegressionWithSGD
data = [
LabeledPoint(-1.0, self.scipy_matrix(2, {1: -1.0})),
LabeledPoint(1.0, self.scipy_matrix(2, {1: 1.0})),