test/py: Add test support for three stage boot