Use trial unittest instead of python unittest