def METHOD_NAME(self):
    print()
    batch_size = 2
    ground = [[4, 5], [6, 7], [8, 9]]
    for idx, nodes in enumerate(
            self.graph.node_batch_iter(
                batch_size=batch_size, shuffle=False, n_type='p')):
        self.assertEqual(len(nodes), batch_size)
        self.assertListEqual(list(nodes), ground[idx])