1.6.11

erdogant · Aug 5, 2023 · 779c29f · 779c29f
1 parent eece445
commit 779c29f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/distfit/__init__.py b/distfit/__init__.py
@@ -2,7 +2,7 @@
 
 __author__ = 'Erdogan Tasksen'
 __email__ = '[email protected]'
-__version__ = '1.6.10'
+__version__ = '1.6.11'
 
 
 # module level doc-string

diff --git a/distfit/tests/test_distfit.py b/distfit/tests/test_distfit.py
@@ -296,13 +296,13 @@ def test_distfit(self):
         dfit = distfit(method='percentile', n_boots=None)
         dfit.fit_transform(X, verbose=0)
         results=dfit.predict(y)
-        assert np.all(np.isin([*results.keys()], ['y', 'y_proba', 'y_pred', 'P', 'teststat']))
+        assert np.all(np.isin([*results.keys()], ['y', 'y_proba', 'y_pred', 'P', 'teststat', 'df']))
 
         # TEST 14: Quantile
         dfit = distfit(method='quantile', n_boots=None)
         dfit.fit_transform(X, verbose=0)
         results=dfit.predict(y)
-        assert np.all(np.isin([*results.keys()],  ['y', 'y_proba', 'y_pred', 'teststat']))
+        assert np.all(np.isin([*results.keys()],  ['y', 'y_proba', 'y_pred', 'teststat', 'df']))
 
         # TEST 15: Discrete
         import random
@@ -322,7 +322,7 @@ def test_distfit(self):
         assert [*dfit.model.keys()]==['name', 'model', 'params', 'score', 'chi2r', 'n', 'p', 'CII_min_alpha', 'CII_max_alpha']
         # TEST 15B
         results = dfit.predict([0, 1, 10, 11, 12])
-        assert np.all(np.isin([*results.keys()], ['y', 'y_proba', 'y_pred', 'P', 'y_bool']))
+        assert np.all(np.isin([*results.keys()], ['y', 'y_proba', 'y_pred', 'P', 'y_bool', 'df']))
 
         from distfit import distfit
         # Set parameters for the test-case