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"])
