Update pymomo unit tests for compatibility with API.