TST Adds more interesting test

thomasjpfan · thomasjpfan · commit effd281000b2 · 2022-04-08T14:34:07.000-04:00
diff --git a/sklearn/utils/tests/test_metaestimators.py b/sklearn/utils/tests/test_metaestimators.py
@@ -94,13 +94,14 @@ def test_if_delegate_has_method():
 class AvailableParameterEstimator:
     """This estimator's `available` parameter toggles the presence of a method"""
 
-    def __init__(self, available=True):
+    def __init__(self, available=True, return_value=1):
         self.available = available
+        self.return_value = return_value
 
     @available_if(lambda est: est.available)
     def available_func(self):
         """This is a mock available_if function"""
-        pass
+        return self.return_value
 
 
 def test_available_if_docstring():
@@ -164,5 +165,8 @@ def test_available_if_methods_can_be_pickled():
 
     Non-regression test for #21344.
     """
-    est = AvailableParameterEstimator(available=True)
-    pickle.dumps(est.available_func)
+    return_value = 10
+    est = AvailableParameterEstimator(available=True, return_value=return_value)
+    pickled_bytes = pickle.dumps(est.available_func)
+    unpickled_func = pickle.loads(pickled_bytes)
+    assert unpickled_func() == return_value
