custom decorator for pytest