Misc test cleanup. Now that scripts are invoked solely through pytest via tox, remove main functions and hash bang lines.

Author: michaelsafyan

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_async_nonstreaming.py

@@ -17,8 +17,6 @@
 # reimplement this in terms of "nonstreaming_base.py".
 
 import asyncio
-import logging
-import unittest
 
 from ..common.base import TestCase
 
@@ -76,12 +74,3 @@ def test_async_generate_content_not_broken_by_instrumentation(self):
             model="gemini-2.0-flash", contents="Does this work?"
         )
         self.assertEqual(response.text, "Yep, it works!")
-
-
-def main():
-    logging.basicConfig(level=logging.DEBUG)
-    unittest.main()
-
-
-if __name__ == "__main__":
-    main()

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_async_streaming.py

@@ -1,5 +1,3 @@
-#!./run_with_env.sh
-
 # Copyright The OpenTelemetry Authors
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,8 +17,6 @@
 # covering the cases of one response and multiple streaming responses.
 
 import asyncio
-import logging
-import unittest
 
 from ..common.base import TestCase
 
@@ -87,12 +83,3 @@ def test_async_generate_content_not_broken_by_instrumentation(self):
         )
         self.assertEqual(len(responses), 1)
         self.assertEqual(responses[0].text, "Yep, it works!")
-
-
-def main():
-    logging.basicConfig(level=logging.DEBUG)
-    unittest.main()
-
-
-if __name__ == "__main__":
-    main()

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_sync_nonstreaming.py

@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import logging
-import unittest
 
 from .nonstreaming_base import NonStreamingTestCase
 
@@ -25,12 +23,3 @@ def generate_content(self, *args, **kwargs):
     @property
     def expected_function_name(self):
         return "google.genai.Models.generate_content"
-
-
-def main():
-    logging.basicConfig(level=logging.DEBUG)
-    unittest.main()
-
-
-if __name__ == "__main__":
-    main()

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_sync_streaming.py

@@ -16,8 +16,6 @@
 # two different tests here that inherit from "streaming_base" and "nonstreaming_base",
 # covering the cases of one response and multiple streaming responses.
 
-import logging
-import unittest
 
 from ..common.base import TestCase
 
@@ -79,12 +77,3 @@ def test_async_generate_content_not_broken_by_instrumentation(self):
         )
         self.assertEqual(len(responses), 1)
         self.assertEqual(responses[0].text, "Yep, it works!")
-
-
-def main():
-    logging.basicConfig(level=logging.DEBUG)
-    unittest.main()
-
-
-if __name__ == "__main__":
-    main()