# third party imports import numpy as np import pkg_resources # local imports from gmprocess.streamcollection import StreamCollection from gmprocess.io.read import read_data from gmprocess.processing import process_streams from gmprocess.logging import setup_logger from gmprocess.io.test_utils import read_data_dir from gmprocess.io.fetch_utils import update_config datapath = os.path.join('data', 'testdata') datadir = pkg_resources.resource_filename('gmprocess', datapath) setup_logger() def test_process_streams(): # Loma Prieta test station (nc216859) data_files, origin = read_data_dir('geonet', 'us1000778i', '*.V1A') streams = [] for f in data_files: streams += read_data(f) sc = StreamCollection(streams) sc.describe() config = update_config(os.path.join(datadir, 'config_min_freq_0p2.yml'))
import os.path import pkg_resources import logging from gmprocess.io.read_directory import directory_to_streams from gmprocess.logging import setup_logger setup_logger() def test_directory_to_streams(): dpath = os.path.join('data', 'testdata', 'read_directory', 'whittier87') directory = pkg_resources.resource_filename('gmprocess', dpath) streams, unprocessed_files, unprocessed_file_errors = directory_to_streams( directory) assert len(streams) == 7 if __name__ == '__main__': os.environ['CALLED_FROM_PYTEST'] = 'True' test_directory_to_streams()