def METHOD_NAME(self):
    pa, pb, pc = [torch.nn.Parameter(data=torch.tensor(i * 1.0)) for i in range(3)]
    na, nb = Node(params=[pa]), Node(params=[pb])
    root = Node(children=[na, nb], params=[pc])
    param_groups = self._get_param_groups(root)
    self._assert_allin([pa, pb, pc], param_groups, "default")