| 21 | |
| 22 | |
| 23 | def _assert_is_identical_nodeflow(nf1, nf2): |
| 24 | assert nf1.num_nodes() == nf2.num_nodes() |
| 25 | src, dst = nf1.all_edges() |
| 26 | src2, dst2 = nf2.all_edges() |
| 27 | assert F.array_equal(src, src2) |
| 28 | assert F.array_equal(dst, dst2) |
| 29 | |
| 30 | assert nf1.num_layers == nf2.num_layers |
| 31 | for i in range(nf1.num_layers): |
| 32 | assert nf1.layer_size(i) == nf2.layer_size(i) |
| 33 | assert nf1.layers[i].data.keys() == nf2.layers[i].data.keys() |
| 34 | for k in nf1.layers[i].data: |
| 35 | assert F.allclose(nf1.layers[i].data[k], nf2.layers[i].data[k]) |
| 36 | assert nf1.num_blocks == nf2.num_blocks |
| 37 | for i in range(nf1.num_blocks): |
| 38 | assert nf1.block_size(i) == nf2.block_size(i) |
| 39 | assert nf1.blocks[i].data.keys() == nf2.blocks[i].data.keys() |
| 40 | for k in nf1.blocks[i].data: |
| 41 | assert F.allclose(nf1.blocks[i].data[k], nf2.blocks[i].data[k]) |
| 42 | |
| 43 | |
| 44 | def _assert_is_identical_batchedgraph(bg1, bg2): |