def test_sort_windows_on_channel_2(): sta_win = {"II.AAK.10.BHZ": [], "II.AAK.10.BHR": [1, 2, 3, 4]} sort_dict = wu.sort_windows_on_channel(sta_win) _true = { "BH": {"nwins": 4, "traces": ["II.AAK.10.BHZ", "II.AAK.10.BHR"]} } assert sort_dict == _true
def test_sort_windows_on_channel(): 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]} sort_dict = wu.sort_windows_on_channel(sta_win) _true = { 'BH': {'nwins': 13, 'traces': ['II.AAK.10.BHR', 'II.AAK..BHZ', 'II.AAK.10.BHT', 'II.AAK.10.BHZ', 'II.AAK.00.BHZ']}, 'EH': {'nwins': 14, 'traces': ['II.AAK..EHR', 'II.AAK.10.EHZ', 'II.AAK..EHZ']}, 'LH': {'nwins': 4, 'traces': ['II.AAK..LHZ']}} assert sort_dict == _true
def test_sort_windows_on_channel(): 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]} sort_dict = wu.sort_windows_on_channel(sta_win) _true = { 'BH': {'nwins': 13, 'traces': ['II.AAK.10.BHR', 'II.AAK..BHZ', 'II.AAK.10.BHT', 'II.AAK.10.BHZ', 'II.AAK.00.BHZ']}, 'EH': {'nwins': 14, 'traces': ['II.AAK..EHR', 'II.AAK.10.EHZ', 'II.AAK..EHZ']}, 'LH': {'nwins': 4, 'traces': ['II.AAK..LHZ']}} for comp in sort_dict: d1 = sort_dict[comp] d2 = _true[comp] assert d1["nwins"] == d2["nwins"] assert set(d1["traces"]) == set(d2["traces"])