Add big-endian data file for LMDB dataset tests
This commit is contained in:
parent
d1002d97ef
commit
d0ed82209b
@ -20,6 +20,7 @@ from __future__ import print_function
|
||||
|
||||
import os
|
||||
import shutil
|
||||
import sys
|
||||
|
||||
from tensorflow.contrib.data.python.ops import readers
|
||||
from tensorflow.python.data.kernel_tests import test_base
|
||||
@ -40,7 +41,10 @@ class LMDBDatasetTest(test_base.DatasetTestBase):
|
||||
def setUp(self):
|
||||
super(LMDBDatasetTest, self).setUp()
|
||||
# Copy database out because we need the path to be writable to use locks.
|
||||
path = os.path.join(prefix_path, "lmdb", "testdata", "data.mdb")
|
||||
# The on-disk format of an LMDB database is different on big-endian
|
||||
# machines, because LMDB is a memory-mapped database.
|
||||
db_file = "data.mdb" if sys.byteorder == "little" else "data_bigendian.mdb"
|
||||
path = os.path.join(prefix_path, "lmdb", "testdata", db_file)
|
||||
self.db_path = os.path.join(self.get_temp_dir(), "data.mdb")
|
||||
shutil.copy(path, self.db_path)
|
||||
|
||||
|
@ -5512,11 +5512,19 @@ filegroup(
|
||||
testonly = 1,
|
||||
srcs = [
|
||||
# A simple key-value store:
|
||||
# 0 : 'b'
|
||||
# 1 : 'b'
|
||||
# ...
|
||||
# 9 : 'b'
|
||||
# Which is then overwritten with:
|
||||
# 0 : 'a'
|
||||
# 1 : 'b'
|
||||
# ...
|
||||
# 9 : 'j'
|
||||
"lib/lmdb/testdata/data.mdb",
|
||||
# LMDB, being a memory-mapped database, uses a different file format on
|
||||
# big-endian systems.
|
||||
"lib/lmdb/testdata/data_bigendian.mdb",
|
||||
],
|
||||
visibility = ["//visibility:public"],
|
||||
)
|
||||
|
BIN
tensorflow/core/lib/lmdb/testdata/data_bigendian.mdb
vendored
Normal file
BIN
tensorflow/core/lib/lmdb/testdata/data_bigendian.mdb
vendored
Normal file
Binary file not shown.
Loading…
Reference in New Issue
Block a user