diff options
| author | Davies Liu <davies.liu@gmail.com> | 2014-10-01 11:21:34 -0700 |
|---|---|---|
| committer | Josh Rosen <joshrosen@apache.org> | 2014-10-01 11:21:34 -0700 |
| commit | abf588f47a26d0066f0b75d52b200a87bb085064 (patch) | |
| tree | d3e9b8a57688805262b7d00c99f596327912ea23 /python/pyspark/tests.py | |
| parent | 0bfd3afb00936b0f46ba613be0982e38bc7032b5 (diff) | |
| download | spark-abf588f47a26d0066f0b75d52b200a87bb085064.tar.gz spark-abf588f47a26d0066f0b75d52b200a87bb085064.tar.bz2 spark-abf588f47a26d0066f0b75d52b200a87bb085064.zip | |
[SPARK-3749] [PySpark] fix bugs in broadcast large closure of RDD
1. broadcast is triggle unexpected
2. fd is leaked in JVM (also leak in parallelize())
3. broadcast is not unpersisted in JVM after RDD is not be used any more.
cc JoshRosen , sorry for these stupid bugs.
Author: Davies Liu <davies.liu@gmail.com>
Closes #2603 from davies/fix_broadcast and squashes the following commits:
080a743 [Davies Liu] fix bugs in broadcast large closure of RDD
Diffstat (limited to 'python/pyspark/tests.py')
| -rw-r--r-- | python/pyspark/tests.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/python/pyspark/tests.py b/python/pyspark/tests.py index 7e2bbc9cb6..6fb6bc998c 100644 --- a/python/pyspark/tests.py +++ b/python/pyspark/tests.py @@ -467,8 +467,12 @@ class TestRDDFunctions(PySparkTestCase): def test_large_closure(self): N = 1000000 data = [float(i) for i in xrange(N)] - m = self.sc.parallelize(range(1), 1).map(lambda x: len(data)).sum() - self.assertEquals(N, m) + rdd = self.sc.parallelize(range(1), 1).map(lambda x: len(data)) + self.assertEquals(N, rdd.first()) + self.assertTrue(rdd._broadcast is not None) + rdd = self.sc.parallelize(range(1), 1).map(lambda x: 1) + self.assertEqual(1, rdd.first()) + self.assertTrue(rdd._broadcast is None) def test_zip_with_different_serializers(self): a = self.sc.parallelize(range(5)) |