Add tests for reshape with an omitted dimension