

def test_mean_probe_position(masks, site_positions):
    masks[:, ::2] *= 0.05
    mean_masks = mean(masks)
    mean_pos = get_mean_probe_position(mean_masks, site_positions)
    assert (mean_pos.shape == (2,))
    assert (mean_pos[0] < 0)
    assert (mean_pos[1] > 0)

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