test.py more pretty (or less ugly)