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()