def gappy_bank(self, gappy_dir):
""" init a sbank on the gappy data """
bank = WaveBank(gappy_dir)
# make sure index is updated after gaps are introduced
if os.path.exists(bank.index_path):
os.remove(bank.index_path)
bank.update_index()
return bank
def default_wbank(tmpdir):
""" create a directory out of the traces in default waveforms, init bank """
base = Path(tmpdir)
st = obspy.read()
for num, tr in enumerate(st):
name = base / f"{(num)}.mseed"
tr.write(str(name), "mseed")
bank = WaveBank(base)
bank.update_index()
return bank
def bank_null_loc_codes(self, tmpdir):
""" create a bank that has nullish location codes in its streams. """
st = obspy.read()
path = Path(tmpdir)
for tr in st:
tr.stats.location = "--"
time = str(get_reference_time(tr))
name = time.split(".")[0].replace(":", "-") + f"_{tr.id}"
tr.write(str(path / name) + ".mseed", "mseed")
bank = WaveBank(path)
bank.update_index()
return bank
def test_put_waveforms_to_crandall_copy(self, tmpdir):
""" ran into issue in docs where putting data into the crandall
copy didn't work. """
ds = obsplus.datasets.utils.copy_dataset(dataset="crandall",
destination=Path(tmpdir))
bank = WaveBank(ds.waveform_client)
bank.read_index() # this sets cache
st = obspy.read()
bank.put_waveforms(st)
bank.update_index()
df = bank.read_index(station="RJOB")
assert len(df) == len(st)
assert set(df.station) == {"RJOB"}
def bank_3(self, tmpdir_factory):
""" Create a bank with several different types of streams. """
td = tmpdir_factory.mktemp("waveforms")
t1, t2 = self.t1, self.t2
bulk3 = [
("TA", "M11A", "01", "CHZ", t1, t2),
("RR", "BOB", "", "HHZ", t1, t2),
("BB", "BOB", "02", "ENZ", t1, t2),
("UU", "SRU", "--", "HHN", t1, t2),
]
ArchiveDirectory(str(td)).create_directory_from_bulk_args(bulk3)
bank = WaveBank(str(td))
bank.update_index()
return bank
def bank(self, df_index, ta_archive):
""" return a bank with monkeypatched index """
sbank = WaveBank(ta_archive)
sbank.update_index = lambda: None
sbank._index_cache = lambda *args, **kwargs: df_index
return sbank
