Function TEST

tensorflow/compiler/xla/text_literal_reader_test.cc:30–54

namespace {

TEST(TextLiteralReaderTest, ReadsR3File) {
  string contents = R"(f32[1,2,3]
(0,0,0): 42.5
(0,0,1): 43.5
(0,0,2): 44.5
(0,1,0): 45.5
(0,1,1): 46.5
(0,1,2): 47.5
)";

  string fname = tensorflow::testing::TmpDir() + "/ReadsR3File.data.txt";
  EXPECT_TRUE(
      tensorflow::WriteStringToFile(tensorflow::Env::Default(), fname, contents)
          .ok());

  Literal literal = TextLiteralReader::ReadPath(fname).ConsumeValueOrDie();
  EXPECT_TRUE(
      ShapeUtil::Equal(ShapeUtil::MakeShape(F32, {1, 2, 3}), literal.shape()));
  EXPECT_EQ(42.5, literal.Get<float>({0, 0, 0}));
  EXPECT_EQ(43.5, literal.Get<float>({0, 0, 1}));
  EXPECT_EQ(44.5, literal.Get<float>({0, 0, 2}));
  EXPECT_EQ(45.5, literal.Get<float>({0, 1, 0}));
  EXPECT_EQ(46.5, literal.Get<float>({0, 1, 1}));
  EXPECT_EQ(47.5, literal.Get<float>({0, 1, 2}));
}

}  // namespace
}  // namespace xla