예제 #1
0
def test_fetch_nyu_rest():
    # First session, all subjects
    nyu = func.fetch_nyu_rest(data_dir=tst.tmpdir, verbose=0)
    assert_equal(len(tst.mock_url_request.urls), 2)
    assert_equal(len(nyu.func), 25)
    assert_equal(len(nyu.anat_anon), 25)
    assert_equal(len(nyu.anat_skull), 25)
    assert_true(np.all(np.asarray(nyu.session) == 1))

    # All sessions, 12 subjects
    tst.mock_url_request.reset()
    nyu = func.fetch_nyu_rest(data_dir=tst.tmpdir,
                              sessions=[1, 2, 3],
                              n_subjects=12,
                              verbose=0)
    # Session 1 has already been downloaded
    assert_equal(len(tst.mock_url_request.urls), 2)
    assert_equal(len(nyu.func), 36)
    assert_equal(len(nyu.anat_anon), 36)
    assert_equal(len(nyu.anat_skull), 36)
    s = np.asarray(nyu.session)
    assert_true(np.all(s[:12] == 1))
    assert_true(np.all(s[12:24] == 2))
    assert_true(np.all(s[24:] == 3))
    assert_not_equal(nyu.description, '')
예제 #2
0
파일: test_func.py 프로젝트: zhiye9/nilearn
def test_fetch_nyu_rest(tmp_path, request_mocker):
    # First session, all subjects
    with pytest.warns(np.VisibleDeprecationWarning,
                      match='fetch_nyu_rest has been deprecated'):
        nyu = func.fetch_nyu_rest(data_dir=str(tmp_path), verbose=0)
    assert request_mocker.url_count == 2
    assert len(nyu.func) == 25
    assert len(nyu.anat_anon) == 25
    assert len(nyu.anat_skull) == 25
    assert np.all(np.asarray(nyu.session) == 1)

    # All sessions, 12 subjects
    with pytest.warns(np.VisibleDeprecationWarning,
                      match='fetch_nyu_rest has been deprecated'):
        nyu = func.fetch_nyu_rest(data_dir=str(tmp_path),
                                  sessions=[1, 2, 3],
                                  n_subjects=12, verbose=0)
    # Session 1 already downloaded, 2b and 3b not needed for 12 subjects
    assert request_mocker.url_count == 4
    assert len(nyu.func) == 36
    assert len(nyu.anat_anon) == 36
    assert len(nyu.anat_skull) == 36
    s = np.asarray(nyu.session)
    assert np.all(s[:12] == 1)
    assert np.all(s[12:24] == 2)
    assert np.all(s[24:] == 3)
    with pytest.warns(np.VisibleDeprecationWarning,
                      match='fetch_nyu_rest has been deprecated'):
        nyu = func.fetch_nyu_rest(data_dir=str(tmp_path),
                                  sessions=[1, 2, 3],
                                  n_subjects=None, verbose=0)
    # For the full dataset (a, b) x (1, 2, 3) = 6 files are downloaded
    assert request_mocker.url_count == 6
    assert nyu.description != ''
예제 #3
0
def test_fetch_nyu_rest(tmp_path, request_mocker):
    # First session, all subjects
    with pytest.warns(np.VisibleDeprecationWarning,
                      match='fetch_nyu_rest has been deprecated'):
        nyu = func.fetch_nyu_rest(data_dir=str(tmp_path), verbose=0)
    assert len(tst.mock_url_request.urls) == 2
    assert len(nyu.func) == 25
    assert len(nyu.anat_anon) == 25
    assert len(nyu.anat_skull) == 25
    assert np.all(np.asarray(nyu.session) == 1)

    # All sessions, 12 subjects
    tst.mock_url_request.reset()
    with pytest.warns(np.VisibleDeprecationWarning,
                      match='fetch_nyu_rest has been deprecated'):
        nyu = func.fetch_nyu_rest(data_dir=str(tmp_path),
                                  sessions=[1, 2, 3],
                                  n_subjects=12,
                                  verbose=0)
    # Session 1 has already been downloaded
    assert len(tst.mock_url_request.urls) == 2
    assert len(nyu.func) == 36
    assert len(nyu.anat_anon) == 36
    assert len(nyu.anat_skull) == 36
    s = np.asarray(nyu.session)
    assert np.all(s[:12] == 1)
    assert np.all(s[12:24] == 2)
    assert np.all(s[24:] == 3)
    assert nyu.description != ''
예제 #4
0
def test_fetch_nyu_rest():
    # First session, all subjects
    nyu = func.fetch_nyu_rest(data_dir=tmpdir, verbose=0)
    assert_equal(len(mock_url_request.urls), 2)
    assert_equal(len(nyu.func), 25)
    assert_equal(len(nyu.anat_anon), 25)
    assert_equal(len(nyu.anat_skull), 25)
    assert_true(np.all(np.asarray(nyu.session) == 1))

    # All sessions, 12 subjects
    mock_url_request.reset()
    nyu = func.fetch_nyu_rest(data_dir=tmpdir, sessions=[1, 2, 3], n_subjects=12, verbose=0)
    # Session 1 has already been downloaded
    assert_equal(len(mock_url_request.urls), 2)
    assert_equal(len(nyu.func), 36)
    assert_equal(len(nyu.anat_anon), 36)
    assert_equal(len(nyu.anat_skull), 36)
    s = np.asarray(nyu.session)
    assert_true(np.all(s[:12] == 1))
    assert_true(np.all(s[12:24] == 2))
    assert_true(np.all(s[24:] == 3))