From 433ac368dfb4d6606646078c1bb9a829dfdbeb34 Mon Sep 17 00:00:00 2001 From: Johnny Robeson Date: Thu, 23 Jun 2016 02:55:59 -0400 Subject: [PATCH] check pull.__next__ on py3 in pipeline tests --- test/test_pipeline.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/test_pipeline.py b/test/test_pipeline.py index 6f95a2208..a40cccdbd 100644 --- a/test/test_pipeline.py +++ b/test/test_pipeline.py @@ -136,7 +136,10 @@ class ExceptionTest(unittest.TestCase): pull = pl.pull() for i in range(3): next(pull) - self.assertRaises(TestException, pull.next) + if six.PY2: + self.assertRaises(TestException, pull.next) + else: + self.assertRaises(TestException, pull.__next__) class ParallelExceptionTest(unittest.TestCase):