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]))