def setUpClass(cls):
cls.default_cache_root = download.get_cache_root()
cls.temp_dir = tempfile.mkdtemp()
download.set_cache_root(cls.temp_dir)
cls.patcher = mock.patch('lineflow.datasets.text_classification.sys.maxsize',
int(sys.float_info.max))
cls.patcher.start()
def test_set_cache_root(self):
orig_root = download.get_cache_root()
new_root = '/tmp/cache'
try:
download.set_cache_root(new_root)
self.assertEqual(download.get_cache_root(), new_root)
finally:
download.set_cache_root(orig_root)
def tearDownClass(cls):
download.set_cache_root(cls.default_cache_root)
shutil.rmtree(cls.temp_dir)
def setUpClass(cls):
cls.default_cache_root = download.get_cache_root()
cls.temp_dir = tempfile.mkdtemp()
download.set_cache_root(cls.temp_dir)
def tearDown(self):
download.set_cache_root(self.default_cache_root)
shutil.rmtree(self.temp_dir)
def setUp(self):
self.default_cache_root = download.get_cache_root()
self.temp_dir = tempfile.mkdtemp()
download.set_cache_root(self.temp_dir)
def tearDown(self):
download.set_cache_root(self.default_cache_root)
os.close(self.temp_file_desc)
os.remove(self.temp_file_name)
shutil.rmtree(self.dir_path)
def setUp(self):
self.default_cache_root = download.get_cache_root()
self.temp_file_desc, self.temp_file_name = tempfile.mkstemp()
download.set_cache_root(self.temp_file_name)
self.dir_path = tempfile.mkdtemp()
