def test_appropriate_entitylist_none_arguments(self):
"""
Test that None arguments are unacceptable
:return:
"""
test_object = EntityListBuilder(None)
with self.assertRaises(AssertionError):
test_object.get_appropriate_entitylist(None)
def test_unranked_is_always_plain_entity_list(self):
"""If the list is unranked, we always get a plain EntityList"""
for option in [
'knn', 'none', 'polynomial', 'exponential', 'svr', 'fake'
]:
test_object = EntityListBuilder(option)
entity_list = test_object.get_appropriate_entitylist(False)
self.assertIsInstance(entity_list, EntityList)
def test_ranked_gives_correct_entity_list_subclass(self, mock_logger):
"""Check that we get the correct EntityList subclass if we have
a ranked list"""
test_object = EntityListBuilder('knn')
entity_list = test_object.get_appropriate_entitylist(True)
self.assertIsInstance(entity_list, KnnEntityList)
test_object = EntityListBuilder('none')
entity_list = test_object.get_appropriate_entitylist(True)
self.assertIsInstance(entity_list, EntityList)
test_object = EntityListBuilder('polynomial')
entity_list = test_object.get_appropriate_entitylist(True)
self.assertIsInstance(entity_list, PolynomialEntityList)
test_object = EntityListBuilder('exponential')
entity_list = test_object.get_appropriate_entitylist(True)
self.assertIsInstance(entity_list, ExponentialEntityList)
test_object = EntityListBuilder('svr')
entity_list = test_object.get_appropriate_entitylist(True)
self.assertIsInstance(entity_list, SvrEntityList)
test_object = EntityListBuilder('fake')
entity_list = test_object.get_appropriate_entitylist(True)
self.assertIsInstance(entity_list, EntityList)
mock_logger.assert_called_with("Unrecognised EntityList Type ("
"'fake'). Returning an unranked "
"EntityList.")
