Model-based testing in Python