Add test case to test_reshape.py