How to write this type of code in vim test.py