Merge pull request #591 from sroet/fix_flaky_test

fix flaky test

test.py

@@ -1493,14 +1493,16 @@ def test_general_signal(self):
 import time
 import signal
 
+i = 0
 def sig_handler(sig, frame):
-    print(10)
-    exit(0)
+    global i
+    i = 42
 
 signal.signal(signal.SIGINT, sig_handler)
 
-for i in range(5):
+for _ in range(6):
     print(i)
+    i += 1
     sys.stdout.flush()
     time.sleep(0.5)
 """)
@@ -1518,7 +1520,7 @@ def agg(line, stdin, process):
         p.wait()
 
         self.assertEqual(p.process.exit_code, 0)
-        self.assertEqual(p, "0\n1\n2\n3\n10\n")
+        self.assertEqual(p, "0\n1\n2\n3\n42\n43\n")
 
     def test_iter_generator(self):
         py = create_tmp_test("""