test diffrent data for svm

ipranjal · Dec 2, 2023 · d22f21e · d22f21e
1 parent bcd7944
commit d22f21e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_api.py b/test_api.py
@@ -23,9 +23,9 @@ def get_processed_data():
 
 def test_predict_svm():
  X,y = get_processed_data()
- response = app.test_client().post("/predict/svm", json={"image":X[0].tolist()})
+ response = app.test_client().post("/predict/svm", json={"image":X[3].tolist()})
  assert response.status_code == 200 
- assert response.get_json()['prediction'] == [0]
+ assert response.get_json()['prediction'] == [3]
 
 def test_predict_tree():
  X,y = get_processed_data()