Esempio n. 1
0
def test_get_plot_stat_map_params_neg():
    # data with negative range
    affine = np.eye(4)
    data_neg = np.array([[-.5, 0, np.nan],
                         [0., np.nan, -.2],
                         [0, 0, 0]])
    img_neg = nibabel.Nifti1Image(data_neg, affine)

    # symmetric_cbar set to True
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=None,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, np.nanmin(data_neg))
    assert_equal(vmax, -np.nanmin(data_neg))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=2,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to False
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=None,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, np.nanmin(data_neg))
    assert_equal(vmax, -np.nanmin(data_neg))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=2,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)

    # symmetric_cbar is set to 'auto', same behaviour as False in this case
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=None,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, np.nanmin(data_neg))
    assert_equal(vmax, -np.nanmin(data_neg))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=2,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)
Esempio n. 2
0
def test_get_plot_stat_map_params_pos():
    # data with positive range
    affine = np.eye(4)
    data_pos = np.array([[0, 1., np.nan],
                         [0., np.nan, 0],
                         [1.5, 2.5, 3.]])
    img_pos = nibabel.Nifti1Image(data_pos, affine)

    # symmetric_cbar set to True
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=None,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data_pos))
    assert_equal(vmax, np.nanmax(data_pos))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=2,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to False
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=None,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data_pos))
    assert_equal(vmax, np.nanmax(data_pos))
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=2,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to 'auto', same behaviour as false in this case
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=None,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data_pos))
    assert_equal(vmax, np.nanmax(data_pos))
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=2,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)
Esempio n. 3
0
def test_get_plot_stat_map_params_pos_neg():
    # data with positive and negative range
    affine = np.eye(4)
    data = np.array([[-.5, 1., np.nan],
                     [0., np.nan, -.2],
                     [1.5, 2.5, 3.]])
    img = nibabel.Nifti1Image(data, affine)

    # symmetric_cbar set to True
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=None,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data))
    assert_equal(vmax, np.nanmax(data))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=2,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to False
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=None,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data))
    assert_equal(vmax, np.nanmax(data))
    assert_equal(cbar_vmin, np.nanmin(data))
    assert_equal(cbar_vmax, np.nanmax(data))
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=2,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, np.nanmin(data))
    assert_equal(cbar_vmax, np.nanmax(data))

    # symmetric_cbar is set to 'auto', same behaviours as True for this case
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=None,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data))
    assert_equal(vmax, np.nanmax(data))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=2,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
Esempio n. 4
0
def test_get_plot_stat_map_params_neg():
    # data with negative range
    affine = np.eye(4)
    data_neg = np.array([[-.5, 0, np.nan], [0., np.nan, -.2], [0, 0, 0]])
    img_neg = nibabel.Nifti1Image(data_neg, affine)

    # symmetric_cbar set to True
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=None,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, np.nanmin(data_neg))
    assert_equal(vmax, -np.nanmin(data_neg))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=2,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to False
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=None,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, np.nanmin(data_neg))
    assert_equal(vmax, -np.nanmin(data_neg))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=2,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)

    # symmetric_cbar is set to 'auto', same behaviour as False in this case
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=None,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, np.nanmin(data_neg))
    assert_equal(vmax, -np.nanmin(data_neg))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_neg, vmax=2,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, 0)
Esempio n. 5
0
def test_get_plot_stat_map_params_pos():
    # data with positive range
    affine = np.eye(4)
    data_pos = np.array([[0, 1., np.nan], [0., np.nan, 0], [1.5, 2.5, 3.]])
    img_pos = nibabel.Nifti1Image(data_pos, affine)

    # symmetric_cbar set to True
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=None,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data_pos))
    assert_equal(vmax, np.nanmax(data_pos))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=2,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to False
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=None,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data_pos))
    assert_equal(vmax, np.nanmax(data_pos))
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=2,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to 'auto', same behaviour as false in this case
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=None,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data_pos))
    assert_equal(vmax, np.nanmax(data_pos))
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img_pos, vmax=2,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, 0)
    assert_equal(cbar_vmax, None)
Esempio n. 6
0
def test_get_plot_stat_map_params_pos_neg():
    # data with positive and negative range
    affine = np.eye(4)
    data = np.array([[-.5, 1., np.nan], [0., np.nan, -.2], [1.5, 2.5, 3.]])
    img = nibabel.Nifti1Image(data, affine)

    # symmetric_cbar set to True
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=None,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data))
    assert_equal(vmax, np.nanmax(data))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=2,
                                  symmetric_cbar=True,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)

    # symmetric_cbar is set to False
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=None,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data))
    assert_equal(vmax, np.nanmax(data))
    assert_equal(cbar_vmin, np.nanmin(data))
    assert_equal(cbar_vmax, np.nanmax(data))
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=2,
                                  symmetric_cbar=False,
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, np.nanmin(data))
    assert_equal(cbar_vmax, np.nanmax(data))

    # symmetric_cbar is set to 'auto', same behaviours as True for this case
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=None,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -np.nanmax(data))
    assert_equal(vmax, np.nanmax(data))
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)
    # same case if vmax has been set
    cbar_vmin, cbar_vmax, vmin, vmax = \
        _get_plot_stat_map_params(img, vmax=2,
                                  symmetric_cbar='auto',
                                  kwargs={})
    assert_equal(vmin, -2)
    assert_equal(vmax, 2)
    assert_equal(cbar_vmin, None)
    assert_equal(cbar_vmax, None)