[Dist] check whether dist partition pipeline works with few num_nodes/edges

[Rhett-Ying]

🔨Work Item

IMPORTANT:

• This template is only for dev team to track project progress. For feature request or bug report, please use the corresponding issue templates.
• DO NOT create a new work item if the purpose is to fix an existing issue or feature request. We will directly use the issue in the project tracker.

Project tracker: https://github.com/orgs/dmlc/projects/2

Description

This is a counterpart of #7779 in distributed partition pipeline.

What needs to do:

1. create a graph with few(less than num_parts) num_nodes/edges of some node/edge types.
2. check if partition goes well.
3. If not, fix it. Refer to [Dist] fix node/edge map for few nodes/edges #7785 about how to fix it.

Depending work items or issues

[Rhett-Ying]

@classicsong FYI