yonikremer
commited on
Commit
•
08b0161
1
Parent(s):
af40811
disabled web search in tests
Browse files
tests.py
CHANGED
@@ -37,15 +37,15 @@ def test_on_form_submit():
|
|
37 |
model_name = "gpt2"
|
38 |
output_length = 10
|
39 |
prompt = "Answer yes or no, is the sky blue?"
|
40 |
-
output = on_form_submit(model_name, output_length, prompt)
|
41 |
assert output is not None
|
42 |
assert len(output) > 0
|
43 |
empty_prompt = ""
|
44 |
with pytest.raises(ValueError):
|
45 |
-
on_form_submit(model_name, output_length, empty_prompt)
|
46 |
unsupported_model_name = "unsupported_model_name"
|
47 |
with pytest.raises(ValueError):
|
48 |
-
on_form_submit(unsupported_model_name, output_length, prompt)
|
49 |
|
50 |
|
51 |
@pytest.mark.parametrize(
|
|
|
37 |
model_name = "gpt2"
|
38 |
output_length = 10
|
39 |
prompt = "Answer yes or no, is the sky blue?"
|
40 |
+
output = on_form_submit(model_name, output_length, prompt, web_search=False)
|
41 |
assert output is not None
|
42 |
assert len(output) > 0
|
43 |
empty_prompt = ""
|
44 |
with pytest.raises(ValueError):
|
45 |
+
on_form_submit(model_name, output_length, empty_prompt, web_search=False)
|
46 |
unsupported_model_name = "unsupported_model_name"
|
47 |
with pytest.raises(ValueError):
|
48 |
+
on_form_submit(unsupported_model_name, output_length, prompt, web_search=False)
|
49 |
|
50 |
|
51 |
@pytest.mark.parametrize(
|