def METHOD_NAME(self):
    max_nodes_per_graph = 10
    h_dense, mask, id = to_dense_batch(
        self.bg.feat,
        batch=None,
        batch_size=None,
        max_num_nodes_per_graph=max_nodes_per_graph,
        drop_nodes_last_graph=False,
    )
    assert mask.size() == (1, max_nodes_per_graph)
    assert torch.sum(mask) == 7
    assert torch.equal(id, torch.arange(7))
    assert h_dense.size() == (1, max_nodes_per_graph, self.bg.feat.size(-1))