def METHOD_NAME(self):
    heads_conf = {"test": {"type": "mlp", "loss": "test_cls"}}
    tasks = ["test"]
    heads_dict = build_heads_dict(heads_conf, tasks, self.losses)
    self.assertTrue(isinstance(heads_dict, HeadsDict))
    # test forward
    task = "test"
    head_output = heads_dict.forward(task, self.model_output, self.sample_list)
    self.assertTrue(isinstance(head_output, dict))
    self.assertTrue("losses" in head_output)
    self.assertTrue("test/test_dataset/logit_bce" in head_output["losses"])