parent
4a081628b2
commit
f381d827d5
@ -1,2 +1,3 @@
|
||||
numpy
|
||||
pyrocko
|
||||
telebot
|
||||
|
@ -1,10 +0,0 @@
|
||||
import os
|
||||
import time
|
||||
import logging
|
||||
import subprocess
|
||||
|
||||
from pyrocko.guts import Bool, Float
|
||||
from .plugin import Plugin, PluginConfig, register_plugin
|
||||
from .meta import Path, DataSize
|
||||
|
||||
logger = logging.getLogger(__name__)
|
@ -0,0 +1,42 @@
|
||||
import numpy as num
|
||||
import pytest
|
||||
|
||||
from hypothesis import given
|
||||
from hypothesis.strategies import floats, integers
|
||||
from hypothesis.extra.numpy import arrays
|
||||
|
||||
|
||||
from pyrocko.trace import Trace
|
||||
|
||||
from idas_convert.idas_convert import split, process_data
|
||||
|
||||
NSAMPLES = 1000
|
||||
|
||||
|
||||
@given(arrays(num.int32, 1000, elements=integers(-1000, 1000)))
|
||||
def test_split(data):
|
||||
tr = Trace(
|
||||
ydata=data,
|
||||
deltat=0.01)
|
||||
tmin_half = (tr.tmax - tr.tmin) / 2
|
||||
traces = split(tr, tmin_half)
|
||||
|
||||
assert len(traces) == 2
|
||||
assert sum(t.ydata.size for t in traces) == tr.ydata.size
|
||||
assert traces[0].tmin == tr.tmin
|
||||
assert traces[1].tmax == tr.tmax
|
||||
|
||||
|
||||
@given(arrays(num.int32, 10000, elements=integers(-1000, 1000)))
|
||||
def test_process_data(data):
|
||||
tr = Trace(
|
||||
ydata=data,
|
||||
deltat=0.001)
|
||||
|
||||
tmin = tr.tmin + .1
|
||||
tmax = tr.tmax - .1
|
||||
deltat = 0.005
|
||||
chunk = (tr, deltat, tmin, tmax)
|
||||
ptr = process_data(chunk)
|
||||
assert ptr.deltat == deltat
|
||||
|
@ -0,0 +1,17 @@
|
||||
from pyrocko.guts import Object
|
||||
|
||||
from idas_convert.meta import DataSize
|
||||
from idas_convert.plugin import Plugin
|
||||
|
||||
def test_data_size():
|
||||
|
||||
class Test(Object):
|
||||
data_size = DataSize.T()
|
||||
|
||||
t = Test(data_size=10000)
|
||||
assert t.data_size == 10000
|
||||
|
||||
|
||||
def test_plugin_base():
|
||||
p = Plugin()
|
||||
p.set_parent('123')
|
Loading…
Reference in new issue