def test_load_svmlight_file():
    X, y = load_svmlight_file(datafile, buffer_mb=1)

    # test X's shape
    assert_equal(X.indptr.shape[0], 4)
    assert_equal(X.shape[0], 3)
    assert_equal(X.shape[1], 21)
    assert_equal(y.shape[0], 3)

    # test X's non-zero values
    for i, j, val in ((0, 2, 2.5), (0, 10, -5.2), (0, 15, 1.5),
                     (1, 5, 1.0), (1, 12, -3),
                     (2, 20, 27)):

        assert_equal(X[i, j], val)

    # tests X's zero values
    assert_equal(X[0, 3], 0)
    assert_equal(X[0, 5], 0)
    assert_equal(X[1, 8], 0)
    assert_equal(X[1, 16], 0)
    assert_equal(X[2, 18], 0)

    # test can change X's values
    X[0, 2] *= 2
    assert_equal(X[0, 2], 5)

    # test y
    assert_array_equal(y, [1, 2, 3])
def test_load_svmlight_file_n_features():
    X, y = load_svmlight_file(datafile, n_features=14, buffer_mb=1)

    # test X'shape
    assert_equal(X.indptr.shape[0], 4)
    assert_equal(X.shape[0], 3)
    assert_equal(X.shape[1], 14)

    # test X's non-zero values
    for i, j, val in ((0, 2, 2.5), (0, 10, -5.2),
                     (1, 5, 1.0), (1, 12, -3)):

        assert_equal(X[i, j], val)
def test_invalid_filename():
    load_svmlight_file("trou pic nic douille")
def test_not_a_filename():
    load_svmlight_file(1)
def test_load_invalid_file():
    X, y = load_svmlight_file(invalidfile, buffer_mb=1)
def test_load_svmlight_file_2_files():
    X_train, y_train, X_test, y_test = load_svmlight_file(datafile,
                                                            datafile)
    assert_array_equal(X_train.toarray(), X_test.toarray())
    assert_array_equal(y_train, y_test)