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)
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)
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)
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)