def test_check_y_pred_1(self): """ Unit test check y pred """ xpl = SmartExplainer() xpl.y_pred = None xpl.x_pred = None xpl.check_y_pred()
def test_check_y_pred_5(self): """ Unit test check y pred 5 """ xpl = SmartExplainer() xpl.x_pred = pd.DataFrame(data=np.array([[1, 2], [3, 4]]), columns=['Col1', 'Col2']) xpl.y_pred = pd.Series(data=np.array(['0'])) with self.assertRaises(ValueError): xpl.check_y_pred(xpl.y_pred)