How to handle ast.Index removal in Python 3.9+ for pytest assertion rewriting?