(idtype)
| 722 | ) |
| 723 | @parametrize_idtype |
| 724 | def test_metis_partition(idtype): |
| 725 | # TODO(zhengda) Metis fails to partition a small graph. |
| 726 | g = create_large_graph(1000, idtype=idtype) |
| 727 | if idtype == F.int64: |
| 728 | check_metis_partition(g, 0) |
| 729 | check_metis_partition(g, 1) |
| 730 | check_metis_partition(g, 2) |
| 731 | check_metis_partition_with_constraint(g) |
| 732 | else: |
| 733 | assert_fail = False |
| 734 | try: |
| 735 | check_metis_partition(g, 1) |
| 736 | except: |
| 737 | assert_fail = True |
| 738 | assert assert_fail |
| 739 | |
| 740 | |
| 741 | def check_metis_partition_with_constraint(g): |
nothing calls this directly
no test coverage detected