tests/test_classifier.py

from language_models_project.main import classify

def testclassifier():
    """ Tests that the classifier classifies as expected """
    test_cases = ["This restaurant is awesome", "This restaurant is awful"]