diff options
| author | Elisey Zanko <elisey.zanko@gmail.com> | 2015-04-20 10:44:09 -0700 |
|---|---|---|
| committer | Josh Rosen <joshrosen@databricks.com> | 2015-04-20 10:44:09 -0700 |
| commit | 77176619a97d07811ab20e1dde4677359d85eb33 (patch) | |
| tree | 3d362532bd4a21d2c888425791418e6001485d80 /python/pyspark/mllib/tests.py | |
| parent | 968ad972175390bb0a96918fd3c7b318d70fa466 (diff) | |
| download | spark-77176619a97d07811ab20e1dde4677359d85eb33.tar.gz spark-77176619a97d07811ab20e1dde4677359d85eb33.tar.bz2 spark-77176619a97d07811ab20e1dde4677359d85eb33.zip | |
[SPARK-6661] Python type errors should print type, not object
Author: Elisey Zanko <elisey.zanko@gmail.com>
Closes #5361 from 31z4/spark-6661 and squashes the following commits:
73c5d79 [Elisey Zanko] Python type errors should print type, not object
Diffstat (limited to 'python/pyspark/mllib/tests.py')
| -rw-r--r-- | python/pyspark/mllib/tests.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py index c6ed5acd17..849c88341a 100644 --- a/python/pyspark/mllib/tests.py +++ b/python/pyspark/mllib/tests.py @@ -135,8 +135,10 @@ class VectorTests(PySparkTestCase): self.assertEquals(sv[-1], 2) self.assertEquals(sv[-2], 0) self.assertEquals(sv[-4], 0) - for ind in [4, -5, 7.8]: + for ind in [4, -5]: self.assertRaises(ValueError, sv.__getitem__, ind) + for ind in [7.8, '1']: + self.assertRaises(TypeError, sv.__getitem__, ind) def test_matrix_indexing(self): mat = DenseMatrix(3, 2, [0, 1, 4, 6, 8, 10]) @@ -450,7 +452,7 @@ class VectorUDTTests(PySparkTestCase): elif isinstance(v, DenseVector): self.assertEqual(v, self.dv1) else: - raise ValueError("expecting a vector but got %r of type %r" % (v, type(v))) + raise TypeError("expecting a vector but got %r of type %r" % (v, type(v))) @unittest.skipIf(not _have_scipy, "SciPy not installed") |