Remove duplicate test (#387)

* Cover methods like _handle_afc

Change-Id: I0f45cb8566b681f5aaeda1500e74d95eaeab10ef

* Remove duplicate code match test

Change-Id: Ia9b7c3dcb8dd4d6bf163303b483225aa0b00e0e9

Author: mayureshagashe2105

tests/test_async_code_match.py

@@ -87,7 +87,7 @@ def test_code_match_for_async_methods(self):
             for node in ast.walk(source_nodes):
                 if isinstance(
                     node, (ast.FunctionDef, ast.AsyncFunctionDef)
-                ) and not node.name.startswith("_"):
+                ) and not node.name.startswith("__"):
                     name = node.name[:-6] if node.name.endswith("_async") else node.name
                     if name in EXEMPT_FUNCTIONS or self._inspect_decorator_exemption(node, fpath):
                         continue

tests/test_generative_models.py

@@ -847,56 +847,6 @@ def test_count_tokens_smoke(self, kwargs):
             {"total_tokens": 7},
         )
 
-    @parameterized.named_parameters(
-        [
-            "GenerateContentResponse",
-            generation_types.GenerateContentResponse,
-            generation_types.AsyncGenerateContentResponse,
-        ],
-        [
-            "GenerativeModel.generate_response",
-            generative_models.GenerativeModel.generate_content,
-            generative_models.GenerativeModel.generate_content_async,
-        ],
-        [
-            "GenerativeModel.count_tokens",
-            generative_models.GenerativeModel.count_tokens,
-            generative_models.GenerativeModel.count_tokens_async,
-        ],
-        [
-            "ChatSession.send_message",
-            generative_models.ChatSession.send_message,
-            generative_models.ChatSession.send_message_async,
-        ],
-        [
-            "ChatSession._handle_afc",
-            generative_models.ChatSession._handle_afc,
-            generative_models.ChatSession._handle_afc_async,
-        ],
-    )
-    def test_async_code_match(self, obj, aobj):
-        import inspect
-        import re
-
-        source = inspect.getsource(obj)
-        asource = inspect.getsource(aobj)
-
-        source = re.sub('""".*"""', "", source, flags=re.DOTALL)
-        asource = re.sub('""".*"""', "", asource, flags=re.DOTALL)
-
-        asource = (
-            asource.replace("anext", "next")
-            .replace("aiter", "iter")
-            .replace("_async", "")
-            .replace("async ", "")
-            .replace("await ", "")
-            .replace("Async", "")
-            .replace("ASYNC_", "")
-        )
-
-        asource = re.sub(" *?# type: ignore", "", asource)
-        self.assertEqual(source, asource, f"error in {obj=}")
-
     def test_repr_for_unary_non_streamed_response(self):
         model = generative_models.GenerativeModel(model_name="gemini-pro")
         self.responses["generate_content"].append(simple_response("world!"))