Suggested changes to test_rotate.py