cosmosmesh-privacy-preservi.../tests/test_adapters.py

25 lines
711 B
Python

import pytest
from cosmosmesh_privacy_preserving_federated_.reference_adapters import (
RoverPlannerAdapter,
HabitatLifeSupportAdapter,
)
def test_rover_adapter_basic():
rover = RoverPlannerAdapter()
assert isinstance(rover.readState(), dict)
data = rover.exposeLocalProblemData()
assert isinstance(data, dict)
resp = rover.applyCommand({"cmd": "test"})
assert resp.get("status") == "accepted"
def test_habitat_adapter_basic():
hab = HabitatLifeSupportAdapter()
assert isinstance(hab.readState(), dict)
data = hab.exposeLocalProblemData()
assert isinstance(data, dict)
resp = hab.applyCommand({"cmd": "test"})
assert resp.get("status") == "accepted"