Work in progress on the upgrade test.
Made upgrade.py try to import both versions of the python driver, it fails though.