test/py: Add base test case for A/B updates