Test functions in test_rotate.py adjusted to work with pytest