wait_for() + partial

I noticed that wait_for does not count with passing a partial without a message. partial() hides the function in its .func parameter.
Merge pull request #469 from mfalesni/wait_partial_improvement

wait_for() + partial