Python assert_sparse_matrix_equal Examples

Example #1

def test_loadtxt():
    X = get_random_coo_matrix()
    f, path = tempfile.mkstemp(suffix='.npz')
    with open(path, 'w') as f:
        for i, j, v in zip(X.row, X.col, X.data):
            print >> f, '{0}\t{1}\t{2}'.format(i + 1, j + 1, v)
    Y = loadtxt(path)
    os.remove(path)
    assert_sparse_matrix_equal(X, Y)

Example #2

File: test_sparse.py Project: ray2020/mrec

def test_loadtxt():
    X = get_random_coo_matrix()
    f, path = tempfile.mkstemp(suffix=".npz")
    with open(path, "w") as f:
        for i, j, v in zip(X.row, X.col, X.data):
            print >> f, "{0}\t{1}\t{2}".format(i + 1, j + 1, v)
    Y = loadtxt(path)
    os.remove(path)
    assert_sparse_matrix_equal(X, Y)

Example #3

File: test_mrec.py Project: ray2020/mrec

def test_save_load_sparse_matrix():
    X = get_random_coo_matrix()
    for fmt in ["tsv", "csv", "npz", "mm", "fsm"]:
        if fmt == "mm":
            suffix = ".mtx"
        elif fmt == "npz" or fmt == "fsm":
            suffix = ".npz"
        else:
            suffix = ""
        f, path = tempfile.mkstemp(suffix=suffix)
        save_sparse_matrix(X, fmt, path)
        Y = load_sparse_matrix(fmt, path)
        assert_sparse_matrix_equal(X, Y)
        os.remove(path)

Example #4

File: test_mrec.py Project: zhouyonglong/mrec-recommender-systems-library

def test_save_load_sparse_matrix():
    X = get_random_coo_matrix()
    for fmt in ['tsv','csv','npz','mm','fsm']:
        if fmt == 'mm':
            suffix = '.mtx'
        elif fmt == 'npz' or fmt == 'fsm':
            suffix = '.npz'
        else:
            suffix = ''
        f,path = tempfile.mkstemp(suffix=suffix)
        save_sparse_matrix(X,fmt,path)
        Y = load_sparse_matrix(fmt,path)
        assert_sparse_matrix_equal(X,Y)
        os.remove(path)