def test_sort_windows_on_channel_and_location_2(): sta_win = {"II.AAK.10.BHZ": [], "II.AAK.00.BHZ": [1, 2, 3]} results = wu.sort_windows_on_channel_and_location(sta_win) _true = {'BH': {'10': {'nwins': 0, 'traces': ['II.AAK.10.BHZ']}, '00': {'nwins': 3, 'traces': ['II.AAK.00.BHZ']}}} assert results == _true sta_win = {} results = wu.sort_windows_on_channel_and_location(sta_win) assert results == {}
def test_sort_windows_on_channel_and_location(): sta_win = {"II.AAK.10.BHZ": [1, 2, 3], "II.AAK.10.BHR": [1, 2], "II.AAK.10.BHT": [1], "II.AAK.00.BHZ": [1, 2, 3, 4], "II.AAK..BHZ": [1, 2, 3], "II.AAK..EHZ": [1, 2, 3, 4, 5, 6, 7, 8], "II.AAK..EHR": [1, 2, 3, 4], "II.AAK.10.EHZ": [1, 2], "II.AAK..LHZ": [1, 2, 3, 4]} results = wu.sort_windows_on_channel_and_location(sta_win) _true = { 'BH': { '10': {'nwins': 6, 'traces': ['II.AAK.10.BHR', 'II.AAK.10.BHT', 'II.AAK.10.BHZ']}, '00': {'nwins': 4, 'traces': ['II.AAK.00.BHZ']}, '': {'nwins': 3, 'traces': ['II.AAK..BHZ']}}, 'EH': { '': {'nwins': 12, 'traces': ['II.AAK..EHR', 'II.AAK..EHZ']}, '10': {'nwins': 2, 'traces': ['II.AAK.10.EHZ']}}, 'LH': { '': {'nwins': 4, 'traces': ["II.AAK..LHZ"]}}} assert results == _true