Function TEST

oneflow/core/ndarray/cpu_concat_var_ndarray_test.cpp:23–33

namespace test {

TEST(CpuConcatVarNdarray, two_elem_concat) {
  std::vector<int32_t> x0_data{0};
  std::vector<int32_t> x1_data{1};
  std::vector<int32_t> buffer{-1, -1};
  std::vector<int32_t> expected{0, 1};
  CpuNdarrayBuilder<int32_t, 1> ndarray;
  auto x0 = ndarray.Var(Shape{1LL}, x0_data.data());
  auto x1 = ndarray.Var(Shape{1LL}, x1_data.data());
  ndarray.Var(Shape{2LL}, buffer.data()).CopyFrom(ndarray.Concatenate({x0, x1}));
  ASSERT_EQ(memcmp(buffer.data(), expected.data(), sizeof(int32_t) * 2), 0);
}

TEST(CpuConcatVarNdarray, two_elem_concat_assign) {
  std::vector<int32_t> x0_data{-1};