def test_class_unmatched(self, mock_classify_article, mock_personQuest, mock_languageQuest, mock_instrumentQuest, mock_cityQuest) : mock_classify_article.return_value = 'abcdef' mock = MagicMock() with mock as sentences : value = ask.getClassification('abcdef', sentences) assert not mock_personQuest.called assert not mock_languageQuest.called assert not mock_instrumentQuest.called assert not mock_cityQuest.called
def test_class_instrument(self, mock_classify_article, mock_instrumentQuest) : mock_classify_article.return_value = 'instrument' mock = MagicMock() with mock as sentences : value = ask.getClassification('instrument', sentences) mock_instrumentQuest.assert_called()
def test_class_city(self, mock_classify_article, mock_cityQuest) : mock_classify_article.return_value = 'city' mock = MagicMock() with mock as sentences : value = ask.getClassification('city', sentences) mock_cityQuest.assert_called()
def test_class_language(self, mock_classify_article, mock_languageQuest) : mock_classify_article.return_value = 'language' mock = MagicMock() with mock as sentences : value = ask.getClassification('language', sentences) mock_languageQuest.assert_called()