Uploaded Test files

venv/Lib/site-packages/sklearn/utils/tests/test_metaestimators.py

@@ -0,0 +1,77 @@
+from sklearn.utils.metaestimators import if_delegate_has_method
+
+
+class Prefix:
+    def func(self):
+        pass
+
+
+class MockMetaEstimator:
+    """This is a mock meta estimator"""
+    a_prefix = Prefix()
+
+    @if_delegate_has_method(delegate="a_prefix")
+    def func(self):
+        """This is a mock delegated function"""
+        pass
+
+
+def test_delegated_docstring():
+    assert "This is a mock delegated function" \
+                in str(MockMetaEstimator.__dict__['func'].__doc__)
+    assert "This is a mock delegated function" \
+           in str(MockMetaEstimator.func.__doc__)
+    assert "This is a mock delegated function" \
+           in str(MockMetaEstimator().func.__doc__)
+
+
+class MetaEst:
+    """A mock meta estimator"""
+    def __init__(self, sub_est, better_sub_est=None):
+        self.sub_est = sub_est
+        self.better_sub_est = better_sub_est
+
+    @if_delegate_has_method(delegate='sub_est')
+    def predict(self):
+        pass
+
+
+class MetaEstTestTuple(MetaEst):
+    """A mock meta estimator to test passing a tuple of delegates"""
+
+    @if_delegate_has_method(delegate=('sub_est', 'better_sub_est'))
+    def predict(self):
+        pass
+
+
+class MetaEstTestList(MetaEst):
+    """A mock meta estimator to test passing a list of delegates"""
+
+    @if_delegate_has_method(delegate=['sub_est', 'better_sub_est'])
+    def predict(self):
+        pass
+
+
+class HasPredict:
+    """A mock sub-estimator with predict method"""
+
+    def predict(self):
+        pass
+
+
+class HasNoPredict:
+    """A mock sub-estimator with no predict method"""
+    pass
+
+
+def test_if_delegate_has_method():
+    assert hasattr(MetaEst(HasPredict()), 'predict')
+    assert not hasattr(MetaEst(HasNoPredict()), 'predict')
+    assert not hasattr(MetaEstTestTuple(HasNoPredict(), HasNoPredict()),
+                       'predict')
+    assert hasattr(MetaEstTestTuple(HasPredict(), HasNoPredict()), 'predict')
+    assert not hasattr(MetaEstTestTuple(HasNoPredict(), HasPredict()),
+                       'predict')
+    assert not hasattr(MetaEstTestList(HasNoPredict(), HasPredict()),
+                       'predict')
+    assert hasattr(MetaEstTestList(HasPredict(), HasPredict()), 'predict')