Is there a reason you are mixing pytest and unittest here? It's not really a problem, just a little messy.