def METHOD_NAME(self):
    pn = op.network.Cubic(shape=[5, 5, 1])
    np.random.seed(10)
    pn.add_model_collection(
        op.models.collections.geometry.spheres_and_cylinders)
    pn.regenerate_models()
    Ts = np.array([0, 4, 6, 18])
    im = op.visualization.plot_connections(pn, throats=Ts,
                                           color_by=pn['throat.diameter'])
    colors_im = im.get_color()
    color_by = pn['throat.diameter'][Ts]
    cscale = (color_by - color_by.min()) / (color_by.max() - color_by.min())
    color_calc = plt.colormaps['jet'](cscale)
    color_calc[:, 3] = 1.0
    assert_allclose(color_calc, colors_im, rtol=1e-5)