def test_random(self): from deepgraph.deepgraph import _triu_indices N = np.random.randint(900, 1100) n = N * (N - 1) / 2 start = np.random.randint(0, n) end = np.random.randint(start, n) indices_true = np.triu_indices(N, k=1) sources_true = indices_true[0][start:end] targets_true = indices_true[1][start:end] indices_test = _triu_indices(N, start, end) sources_test = indices_test[0] targets_test = indices_test[1] assert ((sources_true == sources_test).all() and (targets_true == targets_test).all())
def test_random(self): from deepgraph.deepgraph import _triu_indices N = np.random.randint(900, 1100) n = N*(N-1)/2 start = np.random.randint(0, n) end = np.random.randint(start, n) indices_true = np.triu_indices(N, k=1) sources_true = indices_true[0][start:end] targets_true = indices_true[1][start:end] indices_test = _triu_indices(N, start, end) sources_test = indices_test[0] targets_test = indices_test[1] assert ((sources_true == sources_test).all() and (targets_true == targets_test).all())
def test_border_cases(self): from deepgraph.deepgraph import _triu_indices Ns = [2, 5, 1004, 1523] starts = [0, 0, 0, 9, 9, 9, 9] ends = [0, 1, 2, 9, 10, 11, 12] for N in Ns: for start, end in zip(starts, ends): indices_true = np.triu_indices(N, k=1) sources_true = indices_true[0][start:end] targets_true = indices_true[1][start:end] indices_test = _triu_indices(N, start, end) sources_test = indices_test[0] targets_test = indices_test[1] assert ((sources_true == sources_test).all() and (targets_true == targets_test).all())