def test_extract_sentiment_for_movies(): print("Testing test_extract_sentiment_for_movies() functionality...") chatbot = Chatbot(True) if assertListEquals( chatbot.extract_sentiment_for_movies("I liked both \"I, Robot\" and \"Ex Machina\"."), [("I, Robot", 1), ("Ex Machina", 1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked both \"I, Robot\" and \"Ex Machina\".)\"", orderMatters=False ) and assertListEquals( chatbot.extract_sentiment_for_movies("I liked \"I, Robot\" but not \"Ex Machina\"."), [("I, Robot", 1), ("Ex Machina", -1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked \"I, Robot\" but not \"Ex Machina\".)\"", orderMatters=False ): print('extract_sentiment_for_movies() sanity check passed!') print()
def test_extract_sentiment_for_movies(): print("Testing test_extract_sentiment_for_movies() functionality...") chatbot = Chatbot(True) # add more test cases here!!! test_cases = [ ('I liked both "I, Robot" and "Ex Machina".', [("I, Robot", 1), ("Ex Machina", 1)]), ('I liked "I, Robot" but not "Ex Machina".', [("I, Robot", 1), ("Ex Machina", -1)]), ('I didn\'t like either "I, Robot" or "Ex Machina".', [("I, Robot", -1), ("Ex Machina", -1)]), ('I liked "Titanic (1997)", but "Ex Machina" was not good.', [("Titanic (1997)", 1), ("Ex Machina", -1)]), ] tests_passed = True for input_text, expected_output in test_cases: if not assertListEquals( chatbot.extract_sentiment_for_movies( chatbot.preprocess(input_text)), expected_output, "Incorrect output for extract_sentiment_for_movies(chatbot.preprocess('{}'))." .format(input_text), orderMatters=False): tests_passed = False if tests_passed: print('extract_sentiment_for_movies() sanity check passed!') print()
def test_extract_sentiment_for_movies(): print("Testing test_extract_sentiment_for_movies() functionality...") chatbot = Chatbot(True) chatbot.extract_sentiment_for_movies("I liked both \"I, Robot\" and \"Ex Machina\"") if assertListEquals( chatbot.extract_sentiment_for_movies("I liked both \"I, Robot\" and \"Ex Machina\"."), [("I, Robot", 1), ("Ex Machina", 1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked both \"I, Robot\" and \"Ex Machina\".)\"", orderMatters=False ) and assertListEquals( chatbot.extract_sentiment_for_movies("I liked \"I, Robot\" but not \"Ex Machina\"."), [("I, Robot", 1), ("Ex Machina", -1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked \"I, Robot\" but not \"Ex Machina\".)\"", orderMatters=False ) and assertListEquals( chatbot.extract_sentiment_for_movies("I liked \"I, Robot\" and \"Lady and the Tramp\", but not \"Ex Machina\"."), [("I, Robot", 1), ("Lady and the Tramp", 1), ("Ex Machina", -1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked \"I, Robot\" and \"Lady and the Tramp\", but not \"Ex Machina\".)\"", orderMatters=False ) and assertListEquals( chatbot.extract_sentiment_for_movies("I liked \"I, Robot\", but \"Lady and the Tramp\" was even better! I really liked \"Ex Machina\" too."), [("I, Robot", 1), ("Lady and the Tramp", 1), ("Ex Machina", 1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked \"I, Robot\", but \"Lady and the Tramp\" was even better! I really liked \"Ex Machina\" too.\")\"", orderMatters=False ) and assertListEquals( chatbot.extract_sentiment_for_movies("I liked \"I, Robot\", but the \"Lady and the Tramp\" was not good. But I really liked \"Ex Machina\"."), [("I, Robot", 1), ("Lady and the Tramp", -1), ("Ex Machina", 1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked \"I, Robot\", but the \"Lady and the Tramp\" was not good. But I really liked \"Ex Machina\".\")\"", orderMatters=False ) and assertListEquals( chatbot.extract_sentiment_for_movies("I liked \"I, Robot\", the \"Lady and the Tramp\", and \"Ex Machina\". \"Metropolitan\", however, was quite disappointing."), [("I, Robot", 1), ("Lady and the Tramp", 1), ("Ex Machina", 1), ("Metropolitan", -1)], "Incorrect output for test_extract_sentiment_for_movies(\"I liked \"I, Robot\", the \"Lady and the Tramp\", and \"Ex Machina\". \"Metropolitan\", however, was quite disappointing.\")\"", orderMatters=False ): print('extract_sentiment_for_movies() sanity check passed!') print()