Skip to content

Commit

Permalink
wrap assertRaises with QuietTest
Browse files Browse the repository at this point in the history
  • Loading branch information
megatron-me-uk committed Jun 18, 2015
1 parent 3ab8c7a commit 3344a21
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/pyspark/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -877,7 +877,8 @@ def test_sortByKey_uses_all_partitions_not_only_first_and_last(self):
def test_pipe_functions(self):
data = ['1', '2', '3']
rdd = self.sc.parallelize(data)
self.assertRaises(Exception, rdd.pipe('cc').collect())
with QuietTest(self.sc):
self.assertRaises(Exception, rdd.pipe('cc').collect())
result = rdd.pipe('cat').collect().sort()
[self.assertEqual(x, y) for x, y in zip(data, result)]

Expand Down

0 comments on commit 3344a21

Please sign in to comment.