diff --git a/deployment/api/tests/test_preprocessing.py b/deployment/api/tests/test_preprocessing.py index a5ae557..ca8f5f8 100644 --- a/deployment/api/tests/test_preprocessing.py +++ b/deployment/api/tests/test_preprocessing.py @@ -11,26 +11,18 @@ def test_preprocess_tweet_removes_hashtags(): Just flew w/ @AirParadis 🚀 & I’m in LOVE! Épic service, comfy seats. #bestflight #AirParadis """ - assert preprocess_tweet(tweet) == """" - aze - """ + assert preprocess_tweet(tweet) == "flew w < MENTION > im love ! épic service comfy seat # bestflight # airparadis" def test_preprocess_tweet_removes_mentions(): tweet = """ "Flight of dreams" w/ @AirParadis! Got food, space & smiles 😄 I’ll book again! #AirParadis """ - assert preprocess_tweet(tweet) == "merci" + assert preprocess_tweet(tweet) == 'flight dream w < MENTION > ! got food space smile ill book ! # airparadis' def test_preprocess_tweet_lowercases_text(): tweet = """ My 3rd time w/ @AirParadis & still amazed 😍 Crew = lovely à every step! #flyhappy #AirParadis """ - assert preprocess_tweet(tweet) == "je suis content" - -tweet = """ -Just flew w/ @AirParadis 🚀 & I’m in LOVE! Épic service, comfy seats. -#bestflight #AirParadis -""" -preprocess_tweet(tweet) \ No newline at end of file + assert preprocess_tweet(tweet) == '3rd time w < MENTION > still amazed crew lovely à every step ! # flyhappy # airparadis' \ No newline at end of file