(g1, etype, g, edges_removed)
| 1129 | @parametrize_idtype |
| 1130 | def test_remove_edges(idtype): |
| 1131 | def check(g1, etype, g, edges_removed): |
| 1132 | src, dst, eid = g.edges(etype=etype, form="all") |
| 1133 | src1, dst1 = g1.edges(etype=etype, order="eid") |
| 1134 | if etype is not None: |
| 1135 | eid1 = g1.edges[etype].data[dgl.EID] |
| 1136 | else: |
| 1137 | eid1 = g1.edata[dgl.EID] |
| 1138 | src1 = F.asnumpy(src1) |
| 1139 | dst1 = F.asnumpy(dst1) |
| 1140 | eid1 = F.asnumpy(eid1) |
| 1141 | src = F.asnumpy(src) |
| 1142 | dst = F.asnumpy(dst) |
| 1143 | eid = F.asnumpy(eid) |
| 1144 | sde_set = set(zip(src, dst, eid)) |
| 1145 | |
| 1146 | for s, d, e in zip(src1, dst1, eid1): |
| 1147 | assert (s, d, e) in sde_set |
| 1148 | assert not np.isin(edges_removed, eid1).any() |
| 1149 | assert g1.idtype == g.idtype |
| 1150 | |
| 1151 | for fmt in ["coo", "csr", "csc"]: |
| 1152 | for edges_to_remove in [[2], [2, 2], [3, 2], [1, 3, 1, 2]]: |
no test coverage detected