def _test_frame_read_format(self, format): # test with specific format try: self.frame_read(format=format) except ImportError as e: self.skipTest(str(e)) else: # test again with no format argument # but we need to move other readers out of the way first try: read_ = get_reader('gwf', TimeSeries) except Exception: pass else: register_reader('gwf', TimeSeries, get_reader(format, TimeSeries), force=True) try: self.frame_read() finally: register_reader('gwf', TimeSeries, read_, force=True) # test empty Cache() self.assertRaises(ValueError, self.TEST_CLASS.read, Cache(), self.channel, format=format) # test cache method with `nproc=2` c = Cache.from_urls([TEST_GWF_FILE]) ts = self.TEST_CLASS.read(c, self.channel, nproc=2, format=format)
def make_cache(): segs = SegmentList() cache = Cache() for seg in [(0, 1), (1, 2), (4, 5)]: d = seg[1] - seg[0] f = 'A-B-%d-%d.tmp' % (seg[0], d) cache.append(CacheEntry.from_T050017(f)) segs.append(Segment(*seg)) return cache, segs
def make_cache(): segs = SegmentList() cache = Cache() for seg in [(0, 1), (1, 2), (4, 5)]: d = seg[1] - seg[0] _, f = tempfile.mkstemp(prefix='A-', suffix='-%d-%d.tmp' % (seg[0], d)) cache.append(CacheEntry.from_T050017(f)) segs.append(Segment(*seg)) return cache, segs
def test_io_identify(cls, extensions, modes=['read', 'write']): for mode in modes: for ext in extensions: p = 'X-TEST_CACHE_ENTRY-0-1.%s' % ext c = Cache([CacheEntry.from_T050017(p)]) for path in [p, [p], c[0], c]: formats = identify_format(mode, cls, path, None, [], {}) if len(formats) == 0: raise RuntimeError( "No %s.%s method identified for " "file-format %r in form %r" % (cls.__name__, mode, ext, type(path).__name__)) elif len(formats) != 1: raise RuntimeError( "Multiple %s.%s methods identified for " "file-format %r in form %r" % (cls.__name__, mode, ext, type(path).__name__))