pytest test class