74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
|
|
import os.path
|
|
|
|
from numpy.testing import assert_equal, assert_array_equal
|
|
from nose.tools import raises
|
|
|
|
from sklearn.datasets import load_svmlight_file
|
|
|
|
currdir = os.path.dirname(os.path.abspath(__file__))
|
|
datafile = os.path.join(currdir, "data", "svmlight_classification.txt")
|
|
invalidfile = os.path.join(currdir, "data", "svmlight_invalid.txt")
|
|
|
|
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_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)
|
|
|
|
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)
|
|
|
|
@raises(ValueError)
|
|
def test_load_invalid_file():
|
|
X, y = load_svmlight_file(invalidfile, buffer_mb=1)
|
|
|
|
@raises(TypeError)
|
|
def test_not_a_filename():
|
|
load_svmlight_file(1)
|
|
|
|
@raises(IOError)
|
|
def test_invalid_filename():
|
|
load_svmlight_file("trou pic nic douille")
|