def test_add_3(self):
"""
Unit test add 3
"""
xpl = SmartExplainer()
xpl.columns_dict = {0: 'Age', 1: 'Education', 2: 'Sex'}
xpl.add(features_dict={'Age': 'Age (Years Old)'})
assert xpl.features_dict['Age'] == 'Age (Years Old)'
assert xpl.features_dict['Education'] == 'Education'
def test_add_2(self):
"""
Unit test add 2
"""
xpl = SmartExplainer()
xpl._classes = [0, 1]
xpl._case = "classification"
xpl.add(label_dict={0: 'Zero', 1: 'One'})
assert xpl.label_dict[0] == 'Zero'
assert xpl.label_dict[1] == 'One'
def test_add_1(self, mock_check_y_pred):
"""
Unit test add 1
Parameters
----------
mock_check_y_pred : [type]
[description]
"""
xpl = SmartExplainer()
dataframe_yp = pd.DataFrame([1, 3, 1], columns=['pred'], index=[0, 1, 2])
mock_y_pred = Mock(return_value=dataframe_yp)
mock_check_y_pred.return_value = mock_y_pred()
xpl.x_pred = dataframe_yp
xpl.add(y_pred=dataframe_yp)
expected = SmartExplainer()
expected.y_pred = dataframe_yp
assert not pd.testing.assert_frame_equal(xpl.y_pred, expected.y_pred)
mock_check_y_pred.assert_called()
