Using __main__ method to load unittest