73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
import unittest
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.main import create_app
|
|
|
|
|
|
class WhereAreMoreItemsEndpointTest(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.client = TestClient(create_app())
|
|
|
|
def test_creates_problem(self) -> None:
|
|
response = self.client.post(
|
|
"/math/grade_1/where_are_more_items",
|
|
json={
|
|
"available_pictures": [
|
|
{
|
|
"id": f"picture-{index}",
|
|
"name": f"Picture {index}",
|
|
"image_path": f"/images/{index}.png",
|
|
}
|
|
for index in range(6)
|
|
],
|
|
"seed": 1,
|
|
},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
problem = response.json()
|
|
self.assertEqual(problem["instructions"], "Marca dónde hay más.")
|
|
self.assertEqual(len(problem["comparisons"]), 3)
|
|
|
|
for comparison in problem["comparisons"]:
|
|
left_group = comparison["left_group"]
|
|
right_group = comparison["right_group"]
|
|
|
|
self.assertEqual(left_group["side"], "left")
|
|
self.assertEqual(right_group["side"], "right")
|
|
self.assertNotEqual(left_group["quantity"], right_group["quantity"])
|
|
self.assertIn("picture", left_group)
|
|
self.assertIn("picture", right_group)
|
|
|
|
expected_answer_side = (
|
|
"left"
|
|
if left_group["quantity"] > right_group["quantity"]
|
|
else "right"
|
|
)
|
|
self.assertEqual(comparison["answer_side"], expected_answer_side)
|
|
|
|
def test_returns_bad_request_for_too_few_figures(self) -> None:
|
|
response = self.client.post(
|
|
"/math/grade_1/where_are_more_items",
|
|
json={
|
|
"available_pictures": [
|
|
{
|
|
"id": "picture-1",
|
|
"name": "Picture 1",
|
|
"image_path": "/images/1.png",
|
|
}
|
|
]
|
|
},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertEqual(
|
|
response.json(),
|
|
{"detail": "available_pictures must contain at least 2 pictures"},
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|