refactor: move tests to types directory and add copyright headers

Author: Adewale-1

…models/test_live_client_and_list_type.py …/types/test_live_client_and_list_type.pygoogle/genai/tests/models/test_live_client_and_list_type.py renamed to google/genai/tests/types/test_live_client_and_list_type.py google/genai/tests/models/test_live_client_and_list_type.py renamed to google/genai/tests/types/test_live_client_and_list_type.py

@@ -1,3 +1,18 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 """Tests to verify both LiveClient classes and list[pydantic.BaseModel] support."""
 
 import inspect
@@ -87,11 +102,11 @@ class Recipe(BaseModel):
 
 def test_combined_functionality(client):
     """Test that combines verification of both LiveClient classes and list[pydantic.BaseModel] support."""
-    # 1. Verify LiveClient classes exist
+    # Verify LiveClient classes exist
     assert hasattr(types, "LiveClientMessage")
     assert inspect.isclass(types.LiveClientMessage)
 
-    # 2. Test list[pydantic.BaseModel] support in generate_content
+    # Test the list[pydantic.BaseModel] support in generate_content
     class Recipe(BaseModel):
         recipe_name: str
         ingredients: List[str]

…ai/tests/models/test_parsed_list_mypy.py …nai/tests/types/test_parsed_list_mypy.pygoogle/genai/tests/models/test_parsed_list_mypy.py renamed to google/genai/tests/types/test_parsed_list_mypy.py google/genai/tests/models/test_parsed_list_mypy.py renamed to google/genai/tests/types/test_parsed_list_mypy.py

@@ -1,3 +1,18 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 """Tests to verify that mypy correctly handles list[pydantic.BaseModel] in response.parsed."""
 
 from typing import List, cast
@@ -27,11 +42,11 @@ class Recipe(BaseModel):
     # Create a mock response (simulating what we'd get from the API)
     response = types.GenerateContentResponse()
 
-    # Before the fix, this next line would cause a mypy error:
+    # Before the fix[issue #886], this next line would cause a mypy error:
     # Incompatible types in assignment (expression has type "List[Recipe]",
     # variable has type "Optional[Union[BaseModel, Dict[Any, Any], Enum]]")
     #
-    # With our fix adding list[pydantic.BaseModel] to the Union, this is now valid:
+    # With the fix adding list[pydantic.BaseModel] to the Union, this is now valid:
     response.parsed = [
         Recipe(
             recipe_name="Chocolate Chip Cookies",
@@ -48,16 +63,16 @@ class Recipe(BaseModel):
         # and require a cast:
         # parsed_items = cast(list[Recipe], response.parsed)
 
-        # With our fix, we can directly use it as a list without casting:
+        # With the fix, we can directly use it as a list without casting:
         recipes = response.parsed
 
-        # We can iterate over the list without casting
+        # Now iteration over the list without casting is possible
         for recipe in recipes:
             logger.info(f"Recipe: {recipe.recipe_name}")
             for ingredient in recipe.ingredients:
                 logger.info(f" - {ingredient}")
 
-        # We can access elements by index without casting
+        # Also accessing elements by index without casting is possible
         first_recipe = recipes[0]
         logger.info(f"First recipe: {first_recipe.recipe_name}")
 
@@ -74,12 +89,16 @@ class Recipe(FoodItem):
     response = types.GenerateContentResponse()
 
     # Before the fix, this would require a cast with mypy
-    # Now it works directly with our enhanced type annotation
+    # Now it works directly with the enhanced type annotation
     response.parsed = [
         Recipe(
-            name="Chocolate Chip Cookies", ingredients=["Flour", "Sugar", "Chocolate"]
+            name="Chocolate Chip Cookies",
+            ingredients=["Flour", "Sugar", "Chocolate"],
+        ),
+        Recipe(
+            name="Oatmeal Cookies",
+            ingredients=["Oats", "Flour", "Brown Sugar"],
         ),
-        Recipe(name="Oatmeal Cookies", ingredients=["Oats", "Flour", "Brown Sugar"]),
     ]
 
     if response.parsed is not None:
@@ -157,7 +176,7 @@ class Recipe(BaseModel):
     ]
 
     if response.parsed is not None:
-        # Before our fix, you'd need this cast for mypy to be happy
+        # Before the fix, you'd need this cast for mypy to work successfully
         recipes = cast(List[Recipe], response.parsed)
 
         # Using the cast list

…tests/models/test_parsed_list_support.py …/tests/types/test_parsed_list_support.pygoogle/genai/tests/models/test_parsed_list_support.py renamed to google/genai/tests/types/test_parsed_list_support.py google/genai/tests/models/test_parsed_list_support.py renamed to google/genai/tests/types/test_parsed_list_support.py

@@ -1,3 +1,18 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 import sys
 from enum import Enum
 from typing import List, Optional, Union
@@ -82,8 +97,7 @@ class Recipe(BaseModel):
         recipe_name: str
         ingredients: List[str]
 
-    # Note: This test is artificial since the model would likely return actual recipes,
-    # but we're testing the type annotation support, not the model's behavior
+    # Note: I am only testing the type annotation support, not the model's behavior
 
     # Create a mock response with an empty list
     response = types.GenerateContentResponse()
@@ -120,7 +134,7 @@ class Recipe(BaseModel):
     # Even if the optional fields are None, the type annotation should work
     recipe = response.parsed[0]
     assert isinstance(recipe.recipe_name, str)
-    # Optional fields may or may not be None
+
     assert recipe.prep_time_minutes is None or isinstance(recipe.prep_time_minutes, int)