

def test_sorted_main_channels(masks):
    masks *= 0.05
    masks[:, [5, 7]] *= 20
    mean_masks = mean(masks)
    channels = get_sorted_main_channels(mean_masks, get_unmasked_channels(mean_masks))
    assert np.all(np.in1d(channels, [5, 7]))

dataset/ETHPy150Open kwikteam/phy/phy/stats/tests/test_clusters.py test_sorted_main_channels/original