Add pytest boilerplate code