(g, bg, dst_nodes, include_dst_in_src=True)
| 51 | assert F.array_equal(induced_dst_bg, induced_dst_ans) |
| 52 | |
| 53 | def checkall(g, bg, dst_nodes, include_dst_in_src=True): |
| 54 | for etype in g.etypes: |
| 55 | ntype = g.to_canonical_etype(etype)[2] |
| 56 | if dst_nodes is not None and ntype in dst_nodes: |
| 57 | check(g, bg, ntype, etype, dst_nodes[ntype], include_dst_in_src) |
| 58 | else: |
| 59 | check(g, bg, ntype, etype, None, include_dst_in_src) |
| 60 | |
| 61 | # homogeneous graph |
| 62 | g = dgl.graph( |
no test coverage detected