docstr: Test if get_loss calls the correct loss_function.
docstr_tok: ["Test","if","get_loss","calls","the","correct","loss_function","."]
code: def test_get_loss_call_loss_function(
      self,
      loss_function,
      multilabel_data,
      mock_log_loss,
      mock_multilabel_bce_loss,
      mock_squared_loss,
  ):
      """Test if get_loss calls the correct loss_function."""
      utils.get_loss(
          loss=None,
          labels=np.array([0]),
          logits=np.array([[0.1, -0.1]]),
          probs=None,
          loss_function=loss_function,
          loss_function_using_logits=True,
          multilabel_data=multilabel_data,
      )
      if loss_function in [
          "cross_entropy",
          utils.LossFunction.CROSS_ENTROPY,
      ]:
          if not multilabel_data:
              mock_log_loss.assert_called_once()
          else:
              mock_multilabel_bce_loss.assert_called_once()
      else:
          mock_squared_loss.assert_called_once()
  
imports: from unittest import mock
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from tensorflow_privacy.privacy.privacy_tests import utils
label_desc: Use image node_id 1 for calling the TestGetLoss obj's underlying member method code with example usage: obj.test_get_loss_call_loss_function(loss_function, multilabel_data, mock_log_loss, mock_multilabel_bce_loss, mock_squared_loss) without return types