Update app.py

app.py

@@ -58,6 +58,11 @@ def generate_test(user_code):
 
 ### Instruction:
 You are an expert Python QA engineer. Write a pytest unit test for the following function.
+Rules:
+1. Use the 'pytest' framework.
+2. Do NOT use 'unittest.TestCase' classes.
+3. Use simple functions starting with 'test_'.
+4. Include assert statements.
 
 ### Input:
 {user_code}