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