flaky_decorator_unittest_auto_retry_intermittent_failures.py

1import unittest
2from flaky import flaky
3
4class TestExample(unittest.TestCase):
5    @flaky(max_runs=3, min_passes=1)
6    def test_something_that_is_flaky(self):
7        """
8        This test will be run up to 3 times. 
9        If it succeeds once, the test is considered a success.
10        """
11        import random
12        self.assertTrue(random.choice([True, False]))
13
14if __name__ == '__main__':
15    unittest.main()