Skip to content

Commit e1a8402

Browse files
committed
Update to OpenAI notebook tests
1 parent 731b3bd commit e1a8402

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

test/test_notebook.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,8 @@ def run_notebook(input_nb, output_nb="executed_openai_notebook.ipynb", save=Fals
5454

5555

5656
@pytest.mark.skipif(
57-
skip or not sys.version.startswith("3.10"),
58-
reason="do not run if openai is not installed or py!=3.10",
57+
skip or not sys.version.startswith("3.13"),
58+
reason="do not run if openai is not installed or py!=3.13",
5959
)
6060
def test_agentchat_auto_feedback_from_code(save=False):
6161
run_notebook("agentchat_auto_feedback_from_code_execution.ipynb", save=save)
@@ -78,48 +78,48 @@ def test_agentchat_function_call(save=False):
7878

7979

8080
@pytest.mark.skipif(
81-
skip or not sys.version.startswith("3.13"),
82-
reason="do not run if openai is not installed or py!=3.13",
81+
skip or not sys.version.startswith("3.10"),
82+
reason="do not run if openai is not installed or py!=3.10",
8383
)
8484
def test_agentchat_function_call_currency_calculator(save=False):
8585
run_notebook("agentchat_function_call_currency_calculator.ipynb", save=save)
8686

8787

8888
@pytest.mark.skipif(
89-
skip or not sys.version.startswith("3.10"),
90-
reason="do not run if openai is not installed or py!=3.10",
89+
skip or not sys.version.startswith("3.13"),
90+
reason="do not run if openai is not installed or py!=3.13",
9191
)
9292
def test_agentchat_function_call_async(save=False):
9393
run_notebook("agentchat_function_call_async.ipynb", save=save)
9494

9595

9696
@pytest.mark.skipif(
97-
skip or not sys.version.startswith("3.11"),
98-
reason="do not run if openai is not installed or py!=3.11",
97+
skip or not sys.version.startswith("3.12"),
98+
reason="do not run if openai is not installed or py!=3.12",
9999
)
100100
def _test_agentchat_MathChat(save=False):
101101
run_notebook("agentchat_MathChat.ipynb", save=save)
102102

103103

104104
@pytest.mark.skipif(
105-
skip or not sys.version.startswith("3.12"),
106-
reason="do not run if openai is not installed or py!=3.12",
105+
skip or not sys.version.startswith("3.10"),
106+
reason="do not run if openai is not installed or py!=3.10",
107107
)
108108
def _test_oai_chatgpt_gpt4(save=False):
109109
run_notebook("oai_chatgpt_gpt4.ipynb", save=save)
110110

111111

112112
@pytest.mark.skipif(
113-
skip or not sys.version.startswith("3.13"),
114-
reason="do not run if openai is not installed or py!=3.13",
113+
skip or not sys.version.startswith("3.12"),
114+
reason="do not run if openai is not installed or py!=3.12",
115115
)
116116
def test_agentchat_groupchat_finite_state_machine(save=False):
117117
run_notebook("agentchat_groupchat_finite_state_machine.ipynb", save=save)
118118

119119

120120
@pytest.mark.skipif(
121-
skip or not sys.version.startswith("3.10"),
122-
reason="do not run if openai is not installed or py!=3.10",
121+
skip or not sys.version.startswith("3.11"),
122+
reason="do not run if openai is not installed or py!=3.11",
123123
)
124124
def test_agentchat_cost_token_tracking(save=False):
125125
run_notebook("agentchat_cost_token_tracking.ipynb", save=save)

0 commit comments

Comments
 (0)