Function TEST

tensorflow/lite/kernels/internal/tensor_test.cc:24–29

using ::testing::ElementsAre;

TEST(TensorTest, GetTensorShape4D) {
  RuntimeShape d = GetTensorShape({2, 3, 4, 5});
  EXPECT_THAT(
      std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()),
      ElementsAre(2, 3, 4, 5));
}

TEST(TensorTest, GetTensorShape3D) {
  RuntimeShape d = GetTensorShape({3, 4, 5});