def test_check_explainer_2(self):
"""
Unit test check explainer 2
"""
df = pd.DataFrame(range(0, 21), columns=['id'])
df['y'] = df['id'].apply(lambda x: 1 if x < 10 else 0)
df['x1'] = np.random.randint(1, 123, df.shape[0])
df['x2'] = np.random.randint(1, 3, df.shape[0])
df = df.set_index('id')
clf = cb.CatBoostClassifier(n_estimators=1).fit(df[['x1', 'x2']], df['y'])
explainer = shap.TreeExplainer(clf)
assert explainer.__class__.__base__.__name__ == "Explainer"
check_explainer(explainer)
def check_explainer(self, explainer):
"""
Check if explainer class correspond to a shap explainer object
"""
return check_explainer(explainer)
def test_check_explainer_1(self):
"""
Unit test check explainer 1
"""
explainer = None
assert check_explainer(explainer) == None
def test_check_explainer_1(self):
"""
Unit test check explainer 1
"""
explainer = None
self.assertRaises(ValueError, check_explainer(explainer))