Python wrapの例

コード例 #1

ファイル: test_retry.py プロジェクト: thibault-ml/retrypy

 def test_func_that_raises_too_many_time_raises(self):
     func = retry.wrap(get_dummy_func(50))
     with raises(Exception) as e:
         func()
     assert str(e.value) == 'Test Error 5'

コード例 #2

ファイル: test_retry.py プロジェクト: thibault-ml/retrypy

    def test_retry_func_with_kwargs(self):
        func = retry.wrap(get_dummy_func())

        assert func(foo='bar') == (CallCounter(5, 4), (), {'foo': 'bar'})

コード例 #3

ファイル: test_retry.py プロジェクト: thibault-ml/retrypy

 def test_retry_with_wait_function(self, mock_sleep):
     func = retry.wrap(get_dummy_func(), wait=lambda n: n)
     func()
     mock_sleep.assert_called_with(4)

コード例 #4

ファイル: test_retry.py プロジェクト: thibault-ml/retrypy

    def test_retry_func_with_args(self):
        func = retry.wrap(get_dummy_func())

        assert func('arg') == (CallCounter(5, 4), ('arg', ), {})