def test_loss(self, mock_total_loss, mock_cross):
labels = [[0, 1, 0], [1, 0, 0]]
logits = 'Logits'
smoothing = 0.01
actual = model_train.loss(logits, labels, smoothing)
mock_total_loss.assert_called_once_with()
self.assertEqual(actual, mock_total_loss.return_value)
mock_cross.assert_called_once_with(
logits, labels, label_smoothing=smoothing, weights=1.0)
def test_loss(self, mock_total_loss, mock_cross):
labels = [[0, 1, 0], [1, 0, 0]]
logits = 'Logits'
smoothing = 0.01
actual = model_train.loss(logits, labels, smoothing)
mock_total_loss.assert_called_once_with()
self.assertEqual(actual, mock_total_loss.return_value)
mock_cross.assert_called_once_with(
logits, labels, label_smoothing=smoothing, weights=1.0)
