[MAINT] Add plot_sensors_connectivity data shape check (mne-tools#217)

Co-authored-by: Eric Larson <[email protected]>

mne_connectivity/viz/_3d.py

@@ -57,6 +57,12 @@ def plot_sensors_connectivity(
 
     renderer = _get_renderer(size=(600, 600), bgcolor=(0.5, 0.5, 0.5))
 
+    if con.ndim != 2 or con.shape[0] != con.shape[1]:
+        raise ValueError(
+            "Connectivity data must be a 2D array of shape (n_channels, n_channels), "
+            f"got shape {con.shape}"
+        )
+
     picks = _picks_to_idx(info, picks)
     if len(picks) != len(con):
         raise ValueError(

mne_connectivity/viz/tests/test_3d.py

@@ -27,6 +27,10 @@ def test_plot_sensors_connectivity(renderer):
     info = raw.info
     with pytest.raises(TypeError, match="must be an instance of Info"):
         plot_sensors_connectivity(info="foo", con=con, picks=picks)
+    with pytest.raises(ValueError, match="Connectivity data must be a 2D array"):
+        plot_sensors_connectivity(info=info, con=np.expand_dims(con, 3), picks=picks)
+    with pytest.raises(ValueError, match="2D array of shape (n_channels, n_channels)"):
+        plot_sensors_connectivity(info=info, con=con[:, :-1], picks=picks)
     with pytest.raises(ValueError, match="does not correspond to the size"):
         plot_sensors_connectivity(info=info, con=con[::2, ::2], picks=picks)
     cmap = "viridis"