Mock Committee for testing.
| 79 | |
| 80 | |
| 81 | class MockCommittee: |
| 82 | """ |
| 83 | Mock Committee for testing. |
| 84 | """ |
| 85 | def __init__( |
| 86 | self, n_learners=1, classes_=None, fitted=True, |
| 87 | calculate_disagreement_return=None, |
| 88 | predict_return=None, predict_proba_return=None, |
| 89 | vote_return=None, vote_proba_return=None |
| 90 | ): |
| 91 | self.fitted = fitted |
| 92 | self.n_learners = n_learners |
| 93 | |
| 94 | if fitted: |
| 95 | self.classes_ = classes_ |
| 96 | else: |
| 97 | self.classes_ = None |
| 98 | |
| 99 | self.calculate_disagreement_return = calculate_disagreement_return |
| 100 | self.predict_return = predict_return |
| 101 | self.predict_proba_return = predict_proba_return |
| 102 | self.vote_return = vote_return |
| 103 | self.vote_proba_return = vote_proba_return |
| 104 | |
| 105 | def __len__(self): |
| 106 | return self.n_learners |
| 107 | |
| 108 | def __iter__(self): |
| 109 | for x in range(self.n_learners): |
| 110 | yield x |
| 111 | |
| 112 | def _calculate_disagreement(self, *args, **kwargs): |
| 113 | return self.calculate_disagreement_return |
| 114 | |
| 115 | def predict(self, *args, **kwargs): |
| 116 | if not self.fitted: |
| 117 | raise NotFittedError |
| 118 | |
| 119 | return self.predict_return |
| 120 | |
| 121 | def predict_proba(self, *args, **kwargs): |
| 122 | if not self.fitted: |
| 123 | raise NotFittedError |
| 124 | |
| 125 | return self.predict_proba_return |
| 126 | |
| 127 | def vote(self, *args, **kwargs): |
| 128 | if not self.fitted: |
| 129 | raise NotFittedError |
| 130 | |
| 131 | return self.vote_return |
| 132 | |
| 133 | def vote_proba(self, *args, **kwargs): |
| 134 | if not self.fitted: |
| 135 | raise NotFittedError |
| 136 | |
| 137 | return self.vote_proba_return |
nothing calls this directly
no outgoing calls
no test coverage detected
searching dependent graphs…