def _verify_partition_data_types(part_g):
    """
    check list:
        make sure nodes and edges have correct type.
    """
    ndata = (
        part_g.node_attributes
        if isinstance(part_g, gb.FusedCSCSamplingGraph)
        else part_g.ndata
    )
    edata = (
        part_g.edge_attributes
        if isinstance(part_g, gb.FusedCSCSamplingGraph)
        else part_g.edata
    )

    for k, dtype in RESERVED_FIELD_DTYPE.items():
        if k in ndata:
            assert ndata[k].dtype == dtype
        if k in edata:
            assert edata[k].dtype == dtype

    check list:
        make sure nodes and edges have correct type.
    
tests/distributed/test_partition.py