def test_score(self):
test_cases = (np.random.rand() for _ in range(10))
for score_return in test_cases:
mock_classifier = mock.MockEstimator(score_return=score_return)
learner = modAL.models.ActiveLearner(mock_classifier,
mock.MockFunction(None))
np.testing.assert_almost_equal(
learner.score(np.random.rand(5, 2), np.random.rand(5, )),
score_return)
def test_query(self):
for n_samples in range(1, 100):
for n_features in range(1, 10):
X = np.random.rand(n_samples, n_features)
query_idx = np.random.randint(0, n_samples)
mock_query = mock.MockFunction(return_val=(query_idx,
X[query_idx]))
learner = modAL.models.ActiveLearner(estimator=None,
query_strategy=mock_query)
np.testing.assert_equal(learner.query(X),
(query_idx, X[query_idx]))
