def test_file_list(self): cache = self.make_cache()[0] # test file -> [file.name] with tempfile.NamedTemporaryFile() as f: assert io_cache.file_list(f) == [f.name] # test CacheEntry -> [CacheEntry.path] assert io_cache.file_list(cache[0]) == [cache[0].path] # test cache file -> pfnlist() with tempfile.NamedTemporaryFile(suffix='.lcf', mode='w') as f: io_cache.write_cache(cache, f) f.seek(0) assert io_cache.file_list(f.name) == cache.pfnlist() # test comma-separated list -> list assert io_cache.file_list('A,B,C,D') == ['A', 'B', 'C', 'D'] # test cache object -> pfnlist assert io_cache.file_list(cache) == cache.pfnlist() # test list -> list assert io_cache.file_list(['A', 'B', 'C', 'D']) == ['A', 'B', 'C', 'D'] # otherwise error with pytest.raises(ValueError): io_cache.file_list(1)
def test_file_list(self): cache = self.make_cache()[0] # test file -> [file.name] with tempfile.NamedTemporaryFile() as f: assert io_cache.file_list(f) == [f.name] # test CacheEntry -> [CacheEntry.path] assert io_cache.file_list(cache[0]) == [cache[0].path] # test cache file -> pfnlist() with tempfile.NamedTemporaryFile(suffix='.lcf', mode='w') as f: io_cache.write_cache(cache, f) f.seek(0) assert io_cache.file_list(f.name) == cache.pfnlist() # test comma-separated list -> list assert io_cache.file_list('A,B,C,D') == ['A', 'B', 'C', 'D'] # test cache object -> pfnlist assert io_cache.file_list(cache) == cache.pfnlist() # test list -> list assert io_cache.file_list(['A', 'B', 'C', 'D']) == ['A', 'B', 'C', 'D'] # otherwise error with pytest.raises(ValueError): io_cache.file_list(1)
def test_is_cache_file(self): # check file(path) is return as True if parsed as Cache e = io_cache.CacheEntry.from_T050017('/tmp/A-B-12345-6.txt') with tempfile.NamedTemporaryFile() as f: # empty file should return False assert io_cache.is_cache(f) is False assert io_cache.is_cache(f.name) is False # cache file should return True io_cache.write_cache([e], f) f.seek(0) assert io_cache.is_cache(f) is True assert io_cache.is_cache(f.name) is True
def test_read_write_cache(self): cache = self.make_cache()[0] with tempfile.NamedTemporaryFile() as f: io_cache.write_cache(cache, f) f.seek(0) # read from fileobj c2 = io_cache.read_cache(f) assert cache == c2 # write with file name io_cache.write_cache(cache, f.name) # read from file name c3 = io_cache.read_cache(f.name) assert cache == c3
def test_read_write_cache(self): cache = self.make_cache()[0] with tempfile.NamedTemporaryFile() as f: io_cache.write_cache(cache, f) f.seek(0) # read from fileobj c2 = io_cache.read_cache(f) assert cache == c2 # write with file name io_cache.write_cache(cache, f.name) # read from file name c3 = io_cache.read_cache(f.name) assert cache == c3
def test_is_cache(self): # sanity check assert io_cache.is_cache(None) is False # make sure Cache is returned as True cache = io_cache.Cache() assert io_cache.is_cache(cache) is True # check file(path) is return as True if parsed as Cache cache.append(io_cache.CacheEntry.from_T050017('/tmp/A-B-12345-6.txt')) with tempfile.NamedTemporaryFile() as f: # empty file should return False assert io_cache.is_cache(f) is False assert io_cache.is_cache(f.name) is False # cache file should return True io_cache.write_cache(cache, f) f.seek(0) assert io_cache.is_cache(f) is True assert io_cache.is_cache(f.name) is True # check ASCII file gets returned as False a = numpy.array([[1, 2], [3, 4]]) with tempfile.TemporaryFile() as f: numpy.savetxt(f, a) f.seek(0) assert io_cache.is_cache(f) is False # check HDF5 file gets returned as False try: import h5py except ImportError: pass else: fp = tempfile.mktemp() try: h5py.File(fp, 'w').close() assert io_cache.is_cache(fp) is False finally: if os.path.isfile(fp): os.remove(fp)