wehub-resource-sync
adf0d17497
publish / version_or_publish (push) Has been cancelled
storybook-build / changes (push) Has been cancelled
storybook-build / :storybook-build (push) Has been cancelled
Sync Gradio Skills to Hugging Face / sync-skills (push) Has been cancelled
functional / changes (push) Has been cancelled
functional / build-frontend (push) Has been cancelled
functional / functional-test-SSR=false (push) Has been cancelled
functional / functional-reload (push) Has been cancelled
js / changes (push) Has been cancelled
js / js-test (push) Has been cancelled
docs-build / changes (push) Has been cancelled
docs-build / docs-build (push) Has been cancelled
docs-build / website-build (push) Has been cancelled
functional / functional-test-SSR=true (push) Has been cancelled
hygiene / hygiene-test (push) Has been cancelled
python / changes (push) Has been cancelled
python / build (push) Has been cancelled
python / test-ubuntu-latest-flaky (push) Has been cancelled
python / test-ubuntu-latest-not-flaky (push) Has been cancelled
python / test-windows-latest-flaky (push) Has been cancelled
python / test-windows-latest-not-flaky (push) Has been cancelled
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
from pydantic import BaseModel
|
|
|
|
import gradio as gr
|
|
|
|
|
|
class Person(BaseModel):
|
|
name: str
|
|
age: int
|
|
|
|
|
|
class TestBrowserState:
|
|
def test_preprocess_returns_default_for_none(self):
|
|
state = gr.BrowserState(default_value={"key": "value"})
|
|
assert state.preprocess(None) == {"key": "value"}
|
|
|
|
def test_preprocess_passes_through_payload(self):
|
|
state = gr.BrowserState(default_value="default")
|
|
assert state.preprocess({"name": "Dan", "age": 20}) == {
|
|
"name": "Dan",
|
|
"age": 20,
|
|
}
|
|
|
|
def test_postprocess_passes_through_dict(self):
|
|
state = gr.BrowserState()
|
|
assert state.postprocess({"key": "value"}) == {"key": "value"}
|
|
|
|
def test_postprocess_converts_pydantic_model_to_dict(self):
|
|
state = gr.BrowserState()
|
|
person = Person(name="Dan", age=20)
|
|
result = state.postprocess(person)
|
|
assert result == {"name": "Dan", "age": 20}
|
|
assert isinstance(result, dict)
|
|
|
|
def test_default_value_pydantic_model_converted_to_dict(self):
|
|
state = gr.BrowserState(default_value=Person(name="Dan", age=20))
|
|
assert state.default_value == {"name": "Dan", "age": 20}
|
|
assert isinstance(state.default_value, dict)
|
|
|
|
def test_default_value_plain_types_unchanged(self):
|
|
state_str = gr.BrowserState(default_value="hello")
|
|
assert state_str.default_value == "hello"
|
|
|
|
state_int = gr.BrowserState(default_value=42)
|
|
assert state_int.default_value == 42
|
|
|
|
state_dict = gr.BrowserState(default_value={"a": 1})
|
|
assert state_dict.default_value == {"a": 1}
|
|
|
|
state_none = gr.BrowserState()
|
|
assert state_none.default_value is None
|
|
|
|
def test_preprocess_returns_dict_for_pydantic_default(self):
|
|
"""When default_value is a Pydantic model and payload is None,
|
|
preprocess should return a dict (not a string)."""
|
|
state = gr.BrowserState(default_value=Person(name="Dan", age=20))
|
|
result = state.preprocess(None)
|
|
assert isinstance(result, dict)
|
|
assert result == {"name": "Dan", "age": 20}
|
|
|
|
def test_postprocess_nested_pydantic_model(self):
|
|
class Address(BaseModel):
|
|
city: str
|
|
zip_code: str
|
|
|
|
class PersonWithAddress(BaseModel):
|
|
name: str
|
|
address: Address
|
|
|
|
state = gr.BrowserState()
|
|
person = PersonWithAddress(
|
|
name="Dan", address=Address(city="NYC", zip_code="10001")
|
|
)
|
|
result = state.postprocess(person)
|
|
assert result == {
|
|
"name": "Dan",
|
|
"address": {"city": "NYC", "zip_code": "10001"},
|
|
}
|