1

revdotcom · Dec 21, 2023 · 3949ca3 · 3949ca3
1 parent 33388e9
commit 3949ca3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/tests/fixtures/mock_session.py b/tests/fixtures/mock_session.py
@@ -27,6 +27,6 @@ def _mock_response(url="", status=200, json_data=None, text=""):
         if json_data:
             response.json = mocker.Mock(return_value=json_data)
             type(response).content = mocker.PropertyMock(
-                return_value=json.dumps(json_data).encode('utf-8'))
+                return_value=str(json_data).encode('utf-8'))
         return response
     return _mock_response
diff --git a/tests/test_async_summarization.py b/tests/test_async_summarization.py
@@ -174,7 +174,10 @@ def test_get_transcript_summary_json_paragraph_as_stream(self, mock_session, mak
 
         summary = client.get_transcript_summary_json_as_stream(JOB_ID)
 
-        assert json.loads(summary.content) == data
+        s = summary.content.decode('utf-8')
+        s = s.replace('\'', '"')
+
+        assert json.loads(s) == data
 
         mock_session.request.assert_called_once_with(
             "GET",

diff --git a/tests/test_async_translation.py b/tests/test_async_translation.py
@@ -217,7 +217,9 @@ def test_get_translated_transcript_json_as_stream(self, mock_session, make_mock_
         mock_session.request.return_value = response
 
         translation = client.get_translated_transcript_json_as_stream(JOB_ID, "es")
-        assert json.loads(translation.content) == data
+        s = translation.content.decode('utf-8')
+        s = s.replace('\'', '"')
+        assert json.loads(s) == data
         mock_session.request.assert_called_once_with(
             "GET",
             url,