Simplify unittest usage