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)