Function TEST

dali/kernels/test/block_setup_test.cc:23–31

namespace kernels {

TEST(TensorShape, skip_dim) {
  TensorShape<5> in = { 1, 2, 3, 4, 5 };
  EXPECT_EQ(skip_dim<-1>(in), in);
  EXPECT_EQ(skip_dim<0>(in), TensorShape<4>(   2, 3, 4, 5));  // NOLINT
  EXPECT_EQ(skip_dim<1>(in), TensorShape<4>(1,    3, 4, 5));
  EXPECT_EQ(skip_dim<2>(in), TensorShape<4>(1, 2,    4, 5));
  EXPECT_EQ(skip_dim<3>(in), TensorShape<4>(1, 2, 3,    5));
  EXPECT_EQ(skip_dim<4>(in), TensorShape<4>(1, 2, 3, 4   ));  // NOLINT
}

TEST(BlockSetup, shape2vec) {
  TensorShape<1> in1 = { 123 };