def test_prediction_multiple_langs_with_prob():
lang, p = identify_language('And then he said "Ich liebe dich"!',
with_prob=True, k=2)
assert len(lang) == 2
assert lang[0] == 'en'
assert lang[1] == 'de'
assert len(p) == 2
assert p[0] > 0.5
def test_provide_model_mocked():
model_mock = Mock()
model_mock.predict.return_value = [['__label__elf'], [0.987]]
lang, proba = identify_language('Hello, how are you?', model_mock,
with_prob=True)
model_mock.predict.assert_called_with('Hello, how are you?')
assert lang[0] == 'elf'
assert proba[0] == 0.987
def test_simple_prediction():
lang = identify_language('Hello, how are you?')
assert lang == 'en'