Skip to content

Commit

Permalink
corrected test_app.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Bhuvan588 committed Oct 11, 2024
1 parent c0ee616 commit 5471cac
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions test_app.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def test_index(client):

#Testing predict route when the email content is classified as spam

def test_predict_spam(client, patch):
def test_predict_spam(client, monkeypatch):

def mock_model_predict(features):

Expand All @@ -35,8 +35,8 @@ def mock_vectorizer_transform(text):

return text

patch.setattr('app.model.predict', mock_model_predict)
patch.setattr('app.vectorizer.transform', mock_vectorizer_transform)
monkeypatch.setattr('app.model.predict', mock_model_predict)
monkeypatch.setattr('app.vectorizer.transform', mock_vectorizer_transform)


#Sending a POST request to the /predict route with email content
Expand All @@ -49,7 +49,7 @@ def mock_vectorizer_transform(text):

#Testing predict route when email is classified as non-spam

def test_predict_ham(client, patch):
def test_predict_ham(client, monkeypatch):

def mock_model_predict(features):

Expand All @@ -60,8 +60,8 @@ def mock_vectorizer_transform(text):

return text

patch.setattr('app.model.predict', mock_model_predict)
patch.setattr('app.vectorizer.transform', mock_vectorizer_transform)
monkeypatch.setattr('app.model.predict', mock_model_predict)
monkeypatch.setattr('app.vectorizer.transform', mock_vectorizer_transform)

#Sending a POST request to the /predict route with email content
response = client.post('/predict', data = {'extracted_text': 'This is regarding your job application.'})
Expand Down

0 comments on commit 5471cac

Please sign in to comment.