Add tests for max_pool