Implement more tests based on python end-to-end tests