Files
state-hub/tests/test_hub_core_imports.py

25 lines
908 B
Python

from api.schemas.agent_message import MessageCreate
from api.schemas.doi import DoIReport
from api.schemas.tpsc import GDPR_WARNING_LEVELS, TPSCCatalogRead, TPSCGDPRReport
from hub_core.schemas.agent_message import MessageCreate as CoreMessageCreate
from hub_core.schemas.doi import DoIReport as CoreDoIReport
from hub_core.schemas.tpsc import (
GDPR_WARNING_LEVELS as CORE_GDPR_WARNING_LEVELS,
TPSCCatalogRead as CoreTPSCCatalogRead,
TPSCGDPRReport as CoreTPSCGDPRReport,
)
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
def test_state_hub_reexports_core_tpsc_schemas() -> None:
assert TPSCCatalogRead is CoreTPSCCatalogRead
assert TPSCGDPRReport is CoreTPSCGDPRReport
assert GDPR_WARNING_LEVELS is CORE_GDPR_WARNING_LEVELS