Add a test case for the case where __getstate__ hasn't been overwritten

HolgerPeters · HolgerPeters · commit 9c857f62689c · 2017-02-09T11:23:37.000+01:00
in a second class
diff --git a/sklearn/tests/test_base.py b/sklearn/tests/test_base.py
@@ -386,3 +386,16 @@ def test_pickling_constraints_work_in_multiple_inheritance_setting():
     estimator_restored = pickle.loads(serialized)
     assert estimator_restored.b == 5
     assert estimator_restored._cache is None
+
+
+class SingleInheritanceEstimator(BaseEstimator):
+    def __init__(self, b=5):
+            self.b = b
+
+
+def test_pickling_uses_object_dictionary_when_getstate_not_present():
+    estimator = MultiInheritanceEstimator()
+
+    serialized = pickle.dumps(estimator, protocol=2)
+    estimator_restored = pickle.loads(serialized)
+    assert estimator_restored.b == 5
