Run doctest when calling as main module

Remove test.py