def test_enable_throttling_with_capability(): providers.enable_throttling(delay=4, capability='downloader') assert all([p.request_throttling == 4 and 'downloader' in p.capabilities for p in providers.gets(capability='downloader')]) assert all([p.request_throttling < 4 and not 'downloader' in p.capabilities for p in providers.gets()])
def test_enable_throttling(): providers.enable_throttling(delay=4) assert all([p.request_throttling == 4 for p in providers.gets()])