Function TEST

dali/test/mat2tensor_test.cc:29–38

using kernels::tensor_shape;

TEST(Mat2Tensor, Shape) {
  cv::Mat mat;
  mat.create(480, 640, CV_8UC3);
  EXPECT_EQ(tensor_shape<3>(mat), TensorShape(480, 640, 3));
  mat.create(123, 321, CV_32FC2);
  EXPECT_EQ(tensor_shape<DynamicDimensions>(mat), TensorShape(123, 321, 2));
  mat.create(123, 321, CV_32F);
  EXPECT_EQ(tensor_shape<3>(mat), TensorShape(123, 321, 1));
  EXPECT_EQ(tensor_shape<2>(mat), TensorShape(123, 321));
}


TEST(Mat2Tensor, View) {