Cleanup test module
Improve autocomplete of function docstrings:

- get rid of jedi call on the frontend, this might corrupt the jedi cache (every jedi call should occur on the backend)
- support multiline function definitions