Skip to content

Commit 6e35583

Browse files
committed
Prevent fix for Python 3.9 from breaking tests in other versions.
1 parent 60e09fa commit 6e35583

File tree

1 file changed

+7
-1
lines changed
  • instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content

1 file changed

+7
-1
lines changed

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

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
import json
2929
import os
3030
import subprocess
31+
import sys
3132

3233
import google.auth
3334
import google.auth.credentials
@@ -276,6 +277,11 @@ def _ensure_casette_gzip(loaded_casette):
276277
)
277278

278279

280+
def _maybe_ensure_casette_gzip(result):
281+
if sys.version_info[0] == 3 and sys.version_info[1] == 9:
282+
_ensure_casette_gzip(result)
283+
284+
279285
class _PrettyPrintJSONBody:
280286
"""This makes request and response body recordings more readable."""
281287

@@ -289,7 +295,7 @@ def serialize(cassette_dict):
289295
@staticmethod
290296
def deserialize(cassette_string):
291297
result = yaml.load(cassette_string, Loader=yaml.Loader)
292-
_ensure_casette_gzip(result)
298+
_maybe_ensure_casette_gzip(result)
293299
return result
294300

295301

0 commit comments

Comments
 (0)