fc7775a20f8c4b9f58ee9cbf76516b2b7e823b67,tests/compute/test_nodeflow.py,,test_block_edges,#,351

Before Change

        assert F.array_equal(eid, eid_by_neg)

        dest_nodes = utils.toindex(nf.layer_nid(i + 1))
        u, v, _ = nf._graph.in_edges(dest_nodes)
        u = nf._glb2lcl_nid(u.tousertensor(), i)
        v = nf._glb2lcl_nid(v.tousertensor(), i + 1)
        assert F.array_equal(src, u)
        assert F.array_equal(dst, v)

After Change

        src1 = nf._glb2lcl_nid(src1, i)
        dst1 = nf._glb2lcl_nid(dst1, i + 1)
        assert_array_equal(F.asnumpy(src), F.asnumpy(src1))
        assert_array_equal(F.asnumpy(dst), F.asnumpy(dst1))


def test_block_adj_matrix():
    num_layers = 3

Instances