File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -547,9 +547,9 @@ def test_get_local_property(self):
547
547
"""Verify that local properties set on the driver are available in TaskContext."""
548
548
self .sc .setLocalProperty ("testkey" , "testvalue" )
549
549
rdd = self .sc .parallelize (range (1 ), 1 )
550
- prop1 = rdd1 .map (lambda x : TaskContext .get ().getLocalProperty ("testkey" )).collect ()[0 ]
550
+ prop1 = rdd .map (lambda x : TaskContext .get ().getLocalProperty ("testkey" )).collect ()[0 ]
551
551
self .assertEqual (prop1 , "testkey" )
552
- prop2 = rdd1 .map (lambda x : TaskContext .get ().getLocalProperty ("otherkey" )).collect ()[0 ]
552
+ prop2 = rdd .map (lambda x : TaskContext .get ().getLocalProperty ("otherkey" )).collect ()[0 ]
553
553
self .assertTrue (prop2 is None )
554
554
555
555
class RDDTests (ReusedPySparkTestCase ):
You can’t perform that action at this time.
0 commit comments