Merge pull request statsmodels#5976 from bashtage/restore-resettable-cache

MAINT: Restore ResettableCache

Author: bashtage

MANIFEST.in

@@ -1,4 +1,4 @@
-global-include *.csv *.py *.txt *.pyx *.pyx.in *.pxd *.pxi *.c *.h
+global-include *.csv *.py *.txt *.pyx *.pyx.in *.pxd *.pxi *.c *.h *.pkl
 include MANIFEST.in
 include README.rst
 

setup.py

@@ -128,7 +128,8 @@
     'statsmodels.stats.tests': ['*.txt'],
     'statsmodels.stats.libqsturng': ['*.r', '*.txt', '*.dat'],
     'statsmodels.stats.libqsturng.tests': ['*.csv', '*.dat'],
-    'statsmodels.sandbox.regression.tests': ['*.dta', '*.csv']
+    'statsmodels.sandbox.regression.tests': ['*.dta', '*.csv'],
+    'statsmodels.tsa.statespace.tests.results': ['*.pkl']
 }
 
 ##############################################################################

statsmodels/tools/decorators.py

@@ -3,7 +3,15 @@
 
 import warnings
 
-__all__ = ['cache_readonly', 'cache_writable', 'deprecated_alias']
+__all__ = ['cache_readonly', 'cache_writable', 'deprecated_alias',
+           'ResettableCache']
+
+
+class ResettableCache(dict):
+    """DO NOT USE. BACKWARD COMPAT ONLY"""
+    def __init__(self, *args, **kwargs):
+        super(ResettableCache, self).__init__(*args, **kwargs)
+        self.__dict__ = self
 
 
 def deprecated_alias(old_name, new_name, remove_version=None, msg=None,

statsmodels/tsa/statespace/tests/results/sm-0.9-sarimax.pkl

@@ -1,17 +1,18 @@
 """
 Tests of save / load / remove_data state space functionality.
 """
-
 import pickle
-import tempfile
 import os
+import tempfile
 
 import pytest
 
 from statsmodels import datasets
 from statsmodels.tsa.statespace import (sarimax, structural, varmax,
                                         dynamic_factor)
 from numpy.testing import assert_allclose
+
+current_path = os.path.dirname(os.path.abspath(__file__))
 macrodata = datasets.macrodata.load_pandas().data
 
 
@@ -134,3 +135,9 @@ def test_varmax_pickle(temp_filename):
     assert_allclose(res.params, res2.params)
     assert_allclose(res.bse, res2.bse)
     assert_allclose(res.llf, res2.llf)
+
+
+def test_existing_pickle():
+    pkl_file = os.path.join(current_path, 'results', 'sm-0.9-sarimax.pkl')
+    loaded = sarimax.SARIMAXResults.load(pkl_file)
+    assert isinstance(loaded, sarimax.SARIMAXResultsWrapper)