test.py is now completely unittest and reformated