90 lines
2.0 KiB
Python
90 lines
2.0 KiB
Python
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_prompts():
|
|
return [
|
|
"Hello, my name is",
|
|
"The president of the United States is",
|
|
"The capital of France is",
|
|
"The future of AI is",
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_token_ids():
|
|
return [
|
|
[0],
|
|
[0, 1],
|
|
[0, 2, 1],
|
|
[0, 3, 1, 2],
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_regex():
|
|
return (r"((25[0-5]|(2[0-4]|1\d|[1-9]|)\d)\.){3}"
|
|
r"(25[0-5]|(2[0-4]|1\d|[1-9]|)\d)")
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_json_schema():
|
|
return {
|
|
"type": "object",
|
|
"properties": {
|
|
"name": {
|
|
"type": "string"
|
|
},
|
|
"age": {
|
|
"type": "integer"
|
|
},
|
|
"skills": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "string",
|
|
"maxLength": 10
|
|
},
|
|
"minItems": 3
|
|
},
|
|
"work_history": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "object",
|
|
"properties": {
|
|
"company": {
|
|
"type": "string"
|
|
},
|
|
"duration": {
|
|
"type": "number"
|
|
},
|
|
"position": {
|
|
"type": "string"
|
|
}
|
|
},
|
|
"required": ["company", "position"]
|
|
}
|
|
}
|
|
},
|
|
"required": ["name", "age", "skills", "work_history"]
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_guided_choice():
|
|
return [
|
|
"Python", "Java", "JavaScript", "C++", "C#", "PHP", "TypeScript",
|
|
"Ruby", "Swift", "Kotlin"
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_sql_statements():
|
|
return ("""
|
|
start: select_statement
|
|
select_statement: "SELECT" column "from" table "where" condition
|
|
column: "col_1" | "col_2"
|
|
table: "table_1" | "table_2"
|
|
condition: column "=" number
|
|
number: "1" | "2"
|
|
""")
|