Merge 83db81d001 into 1d434dec47

tests/test_tutorial/test_path_params/test_tutorial004.py

@@ -1,3 +1,4 @@
+import pytest
 from fastapi.testclient import TestClient
 
 from docs_src.path_params.tutorial004 import app
@@ -5,18 +6,18 @@ from docs_src.path_params.tutorial004 import app
 client = TestClient(app)
 
 
-def test_file_path():
-    response = client.get("/files/home/johndoe/myfile.txt")
-    print(response.content)
-    assert response.status_code == 200, response.text
-    assert response.json() == {"file_path": "home/johndoe/myfile.txt"}
+test_data = [
+    ("/files/data/monthly-2024.csv", {"file_path": "data/monthly-2024.csv"}),
+    ("/files/home/johndoe/myfile.txt", {"file_path": "home/johndoe/myfile.txt"}),
+    ("/files//home/johndoe/myfile.txt", {"file_path": "/home/johndoe/myfile.txt"}),
+]
 
 
-def test_root_file_path():
-    response = client.get("/files//home/johndoe/myfile.txt")
-    print(response.content)
-    assert response.status_code == 200, response.text
-    assert response.json() == {"file_path": "/home/johndoe/myfile.txt"}
+@pytest.mark.parametrize("url_path, expected_response", test_data)
+def test_file_paths(url_path, expected_response):
+    response = client.get(url_path)
+    assert response.status_code == 200
+    assert response.json() == expected_response
 
 
 def test_openapi_schema():

tests/test_tutorial/test_path_params/test_tutorial005.py

@@ -1,3 +1,4 @@
+import pytest
 from dirty_equals import IsDict
 from fastapi.testclient import TestClient
 
@@ -6,22 +7,18 @@ from docs_src.path_params.tutorial005 import app
 client = TestClient(app)
 
 
-def test_get_enums_alexnet():
-    response = client.get("/models/alexnet")
-    assert response.status_code == 200
-    assert response.json() == {"model_name": "alexnet", "message": "Deep Learning FTW!"}
-
-
-def test_get_enums_lenet():
-    response = client.get("/models/lenet")
-    assert response.status_code == 200
-    assert response.json() == {"model_name": "lenet", "message": "LeCNN all the images"}
+test_data = [
+    ("/models/alexnet", {"model_name": "alexnet", "message": "Deep Learning FTW!"}),
+    ("/models/lenet", {"model_name": "lenet", "message": "LeCNN all the images"}),
+    ("/models/resnet", {"model_name": "resnet", "message": "Have some residuals"}),
+]
 
 
-def test_get_enums_resnet():
-    response = client.get("/models/resnet")
+@pytest.mark.parametrize("url_path, expected_response", test_data)
+def test_get_enums(url_path, expected_response):
+    response = client.get(url_path)
     assert response.status_code == 200
-    assert response.json() == {"model_name": "resnet", "message": "Have some residuals"}
+    assert response.json() == expected_response
 
 
 def test_get_enums_invalid():