def test_check_contributions_1(self): """ Unit test check contributions 1 """ xpl = SmartExplainer() xpl.contributions, xpl.x_pred = Mock(), Mock() xpl.state = Mock() xpl.check_contributions() xpl.state.check_contributions.assert_called_with(xpl.contributions, xpl.x_pred)
def test_check_contributions_2(self): """ Unit test check contributions 2 """ xpl = SmartExplainer() xpl.contributions, xpl.x_pred = Mock(), Mock() mock_state = Mock() mock_state.check_contributions.return_value = False xpl.state = mock_state with self.assertRaises(ValueError): xpl.check_contributions()