From 60df5f272370e6c26f9586108877e6bde6fcfdba Mon Sep 17 00:00:00 2001
From: YYLIZH <48501277+YYLIZH@users.noreply.github.com>
Date: Tue, 25 Apr 2023 09:20:01 +0800
Subject: [PATCH] fix ci (#257)

---
 tests/test_integration.py | 52 +++++++++++++++++++--------------------
 1 file changed, 26 insertions(+), 26 deletions(-)

diff --git a/tests/test_integration.py b/tests/test_integration.py
index 441484c..ea52be4 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -283,30 +283,30 @@ def test_openai_translate_epub_ja_prompt_json(test_book_dir, tmpdir):
     assert os.path.getsize(os.path.join(tmpdir, "animal_farm_bilingual.epub")) != 0
 
 
-# @pytest.mark.skipif(
-#     not os.environ.get("OPENAI_API_KEY"),
-#     reason="No OPENAI_API_KEY in environment variable.",
-# )
-# def test_openai_translate_srt(test_book_dir, tmpdir):
-#     shutil.copyfile(
-#         os.path.join(test_book_dir, "Lex_Fridman_episode_322.srt"),
-#         os.path.join(tmpdir, "Lex_Fridman_episode_322.srt"),
-#     )
+@pytest.mark.skipif(
+    not os.environ.get("OPENAI_API_KEY"),
+    reason="No OPENAI_API_KEY in environment variable.",
+)
+def test_openai_translate_srt(test_book_dir, tmpdir):
+    shutil.copyfile(
+        os.path.join(test_book_dir, "Lex_Fridman_episode_322.srt"),
+        os.path.join(tmpdir, "Lex_Fridman_episode_322.srt"),
+    )
 
-#     subprocess.run(
-#         [
-#             sys.executable,
-#             "make_book.py",
-#             "--book_name",
-#             os.path.join("Lex_Fridman_episode_322.srt"),
-#             "--test",
-#             "--test_num",
-#             "20",
-#         ],
-#         env=os.environ.copy(),
-#     )
-#     assert os.path.isfile(os.path.join(tmpdir, "Lex_Fridman_episode_322_bilingual.srt"))
-#     assert (
-#         os.path.getsize(os.path.join(tmpdir, "Lex_Fridman_episode_322_bilingual.srt"))
-#         != 0
-#     )
+    subprocess.run(
+        [
+            sys.executable,
+            "make_book.py",
+            "--book_name",
+            os.path.join(tmpdir, "Lex_Fridman_episode_322.srt"),
+            "--test",
+            "--test_num",
+            "20",
+        ],
+        env=os.environ.copy(),
+    )
+    assert os.path.isfile(os.path.join(tmpdir, "Lex_Fridman_episode_322_bilingual.srt"))
+    assert (
+        os.path.getsize(os.path.join(tmpdir, "Lex_Fridman_episode_322_bilingual.srt"))
+        != 0
+    )