MCPcopy Create free account
hub / github.com/modAL-python/modAL / test_set_classes

Method test_set_classes

tests/core_tests.py:1373–1392  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

1371class TestCommittee(unittest.TestCase):
1372
1373 def test_set_classes(self):
1374 # 1. test unfitted learners
1375 for n_learners in range(1, 10):
1376 learner_list = [modAL.models.learners.ActiveLearner(estimator=mock.MockEstimator(fitted=False))
1377 for idx in range(n_learners)]
1378 committee = modAL.models.learners.Committee(
1379 learner_list=learner_list)
1380 self.assertEqual(committee.classes_, None)
1381 self.assertEqual(committee.n_classes_, 0)
1382
1383 # 2. test fitted learners
1384 for n_classes in range(1, 10):
1385 learner_list = [modAL.models.learners.ActiveLearner(estimator=mock.MockEstimator(classes_=np.asarray([idx])))
1386 for idx in range(n_classes)]
1387 committee = modAL.models.learners.Committee(
1388 learner_list=learner_list)
1389 np.testing.assert_equal(
1390 committee.classes_,
1391 np.unique(range(n_classes))
1392 )
1393
1394 def test_predict(self):
1395 for n_learners in range(1, 10):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected