Update mne_connectivity/viz/tests/test_3d.py

Co-authored-by: Daniel McCloy <dan@mccloy.info>
mne-tools · Jul 15, 2023 · 12c600c · 12c600c
1 parent e4e66d6
commit 12c600c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mne_connectivity/viz/tests/test_3d.py b/mne_connectivity/viz/tests/test_3d.py
@@ -37,7 +37,8 @@ def test_plot_sensors_connectivity(renderer):
     cmap_from_vtk = np.array(
         fig.plotter.scalar_bar.GetLookupTable().GetTable()
     )
-    cmap_from_vtk = cmap_from_vtk[:, :3] / cmap_from_vtk[:, [-1]]
+    # discard alpha channel and convert uint8 -> norm
+    cmap_from_vtk = cmap_from_vtk[:, :3] / 255
     cmap_from_vtk = cmap_from_vtk[::-1]  # for some reason order is flipped
     assert_almost_equal(cmap_from_mpl, cmap_from_vtk, decimal=2)
     # check title