Files
state-hub/tests/test_hub_core_imports.py

13 lines
432 B
Python

from api.schemas.agent_message import MessageCreate
from api.schemas.doi import DoIReport
from hub_core.schemas.agent_message import MessageCreate as CoreMessageCreate
from hub_core.schemas.doi import DoIReport as CoreDoIReport
def test_state_hub_reexports_core_message_schema() -> None:
assert MessageCreate is CoreMessageCreate
def test_state_hub_reexports_core_doi_schema() -> None:
assert DoIReport is CoreDoIReport