/**
 * \file dnn/test/naive/images2neibs.cpp
 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
 *
 * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 */
#include "test/naive/fixture.h"

#include "megdnn/oprs/nn.h"
#include "test/common/checker.h"

using namespace megdnn;
using namespace test;

TEST_F(NAIVE, IMAGES2NEIBS_FORWARD) {
    Checker<Images2Neibs> checker(handle(), /* check_dispatch */false);
    
    Images2Neibs::Param param(0,0,1,1,1,1,2,2);
    checker.set_param(param).exect(
            Testcase{TensorValue({1, 1, 3, 3}, dtype::Uint8(),
                               {0,1,2,
                               3,4,5,
                               6,7,8}), {}}, 
            Testcase{{},
            TensorValue({1, 1, 2, 2, 2, 2}, dtype::Uint8(), 
                        {0,1,3,4,
                        1,2,4,5,
                        3,4,6,7,
                        4,5,7,8})});
    
    param.pad_h = 1;
    param.pad_w = 1;
    param.stride_h = 2;
    param.stride_w = 2;
    param.dilate_h = 2;
    param.dilate_w = 2;
    param.window_h = 3;
    param.window_w = 3;
    checker.set_param(param).exect(
          Testcase{TensorValue({1, 1, 6, 7}, dtype::Uint8(),
                               {0,1,2,3,4,5,6,
                               7,8,9,10,11,12,13,
                               14,15,16,17,18,19,20,
                               21,22,23,24,25,26,27,
                               28,29,30,31,32,33,34,
                               35,36,37,38,39,40,41}), {}}, 
          Testcase{{},
          TensorValue({1, 1, 2, 3, 3, 3}, dtype::Uint8(),
                        {0,0,0,0,8,10,0,22,24,
                        0,0,0,8,10,12,22,24,26,
                        0,0,0,10,12,0,24,26,0,
                        0,8,10,0,22,24,0,36,38,
                        8,10,12,22,24,26,36,38,40,
                        10,12,0,24,26,0,38,40,0})});
}