def METHOD_NAME(self):
    tags, keypoint_indices = self._make_input(num_instance=2)
    # test loss calculation
    loss_module = AssociativeEmbeddingLoss()
    pull_loss, push_loss = loss_module(tags, keypoint_indices)
    _pull_loss, _push_loss = AELoss('exp')(tags, keypoint_indices)
    self.assertTrue(torch.allclose(pull_loss, _pull_loss))
    self.assertTrue(torch.allclose(push_loss, _push_loss))
    # test loss weight
    loss_module = AssociativeEmbeddingLoss(loss_weight=0.)
    pull_loss, push_loss = loss_module(tags, keypoint_indices)
    self.assertTrue(torch.allclose(pull_loss, torch.zeros(1)))
    self.assertTrue(torch.allclose(push_loss, torch.zeros(1)))
    # test push loss factor
    loss_module = AssociativeEmbeddingLoss(push_loss_factor=0.)
    pull_loss, push_loss = loss_module(tags, keypoint_indices)
    self.assertFalse(torch.allclose(pull_loss, torch.zeros(1)))
    self.assertTrue(torch.allclose(push_loss, torch.zeros(1)))