aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/util/memmapped_file_system_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/util/memmapped_file_system_test.cc')
-rw-r--r--tensorflow/core/util/memmapped_file_system_test.cc8
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/util/memmapped_file_system_test.cc b/tensorflow/core/util/memmapped_file_system_test.cc
index cf03647264..a209ef8b9b 100644
--- a/tensorflow/core/util/memmapped_file_system_test.cc
+++ b/tensorflow/core/util/memmapped_file_system_test.cc
@@ -90,6 +90,11 @@ TEST(MemmappedFileSystemTest, SimpleTest) {
TF_ASSERT_OK(memmapped_env.GetFileSize(kTensor2FileName, &file_size));
EXPECT_EQ(test_tensor.TotalBytes(), file_size);
+ // Check that Stat works.
+ FileStatistics stat;
+ TF_ASSERT_OK(memmapped_env.Stat(kTensor2FileName, &stat));
+ EXPECT_EQ(test_tensor.TotalBytes(), stat.length);
+
// Check that if file not found correct error message returned.
EXPECT_EQ(
error::NOT_FOUND,
@@ -139,6 +144,9 @@ TEST(MemmappedFileSystemTest, ProxyToDefault) {
uint64 file_length = 0;
TF_EXPECT_OK(memmapped_env.GetFileSize(filename, &file_length));
EXPECT_EQ(test_string.length(), file_length);
+ FileStatistics stat;
+ TF_EXPECT_OK(memmapped_env.Stat(filename, &stat));
+ EXPECT_EQ(test_string.length(), stat.length);
std::unique_ptr<RandomAccessFile> random_access_file;
TF_ASSERT_OK(
memmapped_env.NewRandomAccessFile(filename, &random_access_file));