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)