def test_classify_nn():
    """Test classify using nn"""
    classifer = ReviewClassifier('nn')
    classifer.load_model(tar_file='test_nn_model.tar')
    classifer.classify('classified_comments.txt',
                       comments_filename='test-reviews.csv')
    assert os.path.exists('classified_comments.txt')
    os.remove('classified_comments.txt')
def test_classify_nb_from_csv():
    """Test classify using nb model from csv file"""
    classifer = ReviewClassifier('nb')
    classifer.load_model(pickle_file='test_nb_model.pickle')
    classifer.classify('classified_comments.txt',
                       comments_filename='test-reviews.csv')
    assert os.path.exists('classified_comments.txt')
    os.remove('classified_comments.txt')
def test_classify_rf():
    """Test classify using rf"""
    classifer = ReviewClassifier('rf')
    classifer.load_model(pickle_file='test_rf_model.pickle')
    classifer.classify('classified_comments.txt',
                       comments_filename='test-reviews.csv')
    assert os.path.exists('classified_comments.txt')
    os.remove('classified_comments.txt')
def test_classify_from_text_file():
    """Test classify comments text file"""
    classifer = ReviewClassifier('nb')
    classifer.load_model(pickle_file='test_nb_model.pickle')
    classifer.classify('classified_comments.txt',
                       comments_text_file='neutral.txt')
    assert os.path.exists('classified_comments.txt')
    os.remove('classified_comments.txt')