Python flex_cut_traceの例

コード例 #1

ファイル: test_process.py プロジェクト: wjlei1990/pytomo3d

def test_flex_cut_trace():

    st = obspy.read(small_mseed)
    tr_old = st[0]
    tstart = tr_old.stats.starttime
    tend = tr_old.stats.endtime
    npts = tr_old.stats.npts
    dt = tr_old.stats.delta

    tr = tr_old.copy()
    t1 = tstart + int(npts / 4) * dt
    t2 = tend - int(npts / 4) * dt
    proc.flex_cut_trace(tr, t1, t2)
    assert tr.stats.starttime == t1
    assert tr.stats.endtime == t2

    tr = tr_old.copy()
    t1 = tstart + 20 * dt
    t2 = tend - 20 * dt
    proc.flex_cut_trace(tr, t1, t2, dynamic_npts=10)
    assert tr.stats.starttime == (t1 - 10 * dt)
    assert tr.stats.endtime == (t2 + 10 * dt)

    tr = tr_old.copy()
    t1 = tstart - int(npts / 4) * dt
    t2 = tend + int(npts / 4) * dt
    proc.flex_cut_trace(tr, t1, t2)
    assert tr.stats.starttime == tstart
    assert tr.stats.endtime == tend

    tr = tr_old.copy()
    t1 = tstart + int(npts * 0.8) * dt
    t2 = tend - int(npts * 0.8) * dt
    with pytest.raises(ValueError):
        proc.flex_cut_trace(tr, t1, t2)

コード例 #2

ファイル: test_process.py プロジェクト: wjlei1990/pytomo3d

def test_flex_cut_trace():

    st = obspy.read(small_mseed)
    tr_old = st[0]
    tstart = tr_old.stats.starttime
    tend = tr_old.stats.endtime
    npts = tr_old.stats.npts
    dt = tr_old.stats.delta

    tr = tr_old.copy()
    t1 = tstart + int(npts / 4) * dt
    t2 = tend - int(npts / 4) * dt
    proc.flex_cut_trace(tr, t1, t2)
    assert tr.stats.starttime == t1
    assert tr.stats.endtime == t2

    tr = tr_old.copy()
    t1 = tstart + 20 * dt
    t2 = tend - 20 * dt
    proc.flex_cut_trace(tr, t1, t2, dynamic_npts=10)
    assert tr.stats.starttime == (t1 - 10 * dt)
    assert tr.stats.endtime == (t2 + 10 * dt)

    tr = tr_old.copy()
    t1 = tstart - int(npts / 4) * dt
    t2 = tend + int(npts / 4) * dt
    proc.flex_cut_trace(tr, t1, t2)
    assert tr.stats.starttime == tstart
    assert tr.stats.endtime == tend

    tr = tr_old.copy()
    t1 = tstart + int(npts * 0.8) * dt
    t2 = tend - int(npts * 0.8) * dt
    with pytest.raises(ValueError):
        proc.flex_cut_trace(tr, t1, t2)

コード例 #3

ファイル: test_signal.py プロジェクト: chukren/pytomo3d

def test_flex_cut_trace():
    st = obspy.read(small_mseed)
    tr = st[0]
    tstart = tr.stats.starttime
    tend = tr.stats.endtime
    npts = tr.stats.npts
    dt = tr.stats.delta

    t1 = tstart + int(npts / 4) * dt
    t2 = tend - int(npts / 4) * dt
    tr_cut = proc.flex_cut_trace(tr, t1, t2)
    assert tr_cut.stats.starttime == t1
    assert tr_cut.stats.endtime == t2

    t1 = tstart - int(npts / 4) * dt
    t2 = tend + int(npts / 4) * dt
    tr_cut = proc.flex_cut_trace(tr, t1, t2)
    assert tr_cut.stats.starttime == tstart
    assert tr_cut.stats.endtime == tend

    t1 = tstart + int(npts * 0.8) * dt
    t2 = tend - int(npts * 0.8) * dt
    with pytest.raises(ValueError):
        proc.flex_cut_trace(tr, t1, t2)