Increase Java heap size for Python tests