def setUp(self):
super(TestCase, self).setUp()
self.patchers = []
# New data_dir and builder for each test
self.data_dir = test_utils.make_tmp_dir(self.get_temp_dir())
self.builder = self._make_builder()
self.example_dir = os.path.join(
os.path.dirname(__file__),
"test_data/fake_examples/%s" % self.builder.name)
if self.MOCK_OUT_FORBIDDEN_OS_FUNCTIONS:
self._mock_out_forbidden_os_functions()
def setUpClass(cls):
tmp_dir = test_utils.make_tmp_dir(tf.test.get_temp_dir())
inner_dir = os.path.join(tmp_dir, "foo")
tf.gfile.MakeDirs(inner_dir)
dummy_fname = os.path.join(inner_dir, "dummy.txt")
file_contents = "hello world"
with tf.gfile.Open(dummy_fname, "w") as f:
f.write(file_contents)
cls.tmp_dir = tmp_dir
cls.inner_dir = inner_dir
cls.dummy_fname = dummy_fname
cls.file_contents = file_contents
def setUp(self):
super(TestCase, self).setUp()
self.patchers = []
# New data_dir and builder for each test
self.data_dir = test_utils.make_tmp_dir(self.get_temp_dir())
self.builder = self._make_builder()
self.example_dir = os.path.join(
os.path.dirname(__file__),
"test_data/fake_examples/%s" % self.builder.name)
if not tf.io.gfile.exists(self.example_dir):
err_msg = "fake_examples dir %s not found." % self.example_dir
raise ValueError(err_msg)
if self.MOCK_OUT_FORBIDDEN_OS_FUNCTIONS:
self._mock_out_forbidden_os_functions()
def setUpClass(cls): cls._tfds_tmp_dir = test_utils.make_tmp_dir() builder = DummyDatasetSharedGenerator(data_dir=cls._tfds_tmp_dir) builder.download_and_prepare()
def setUpClass(cls):
cls._builder = DummyDataset(data_dir=test_utils.make_tmp_dir())
cls._builder.download_and_prepare()
def setUpClass(cls):
cls._tfds_tmp_dir = test_utils.make_tmp_dir()
cls._builder = DummyDatasetSharedGenerator(data_dir=cls._tfds_tmp_dir)
def setUpClass(cls):
super(DatasetInfoTest, cls).setUpClass()
cls._tfds_tmp_dir = test_utils.make_tmp_dir()
cls._builder = DummyDatasetSharedGenerator(data_dir=cls._tfds_tmp_dir)
