mirror of
https://github.com/tiqi-group/pydase.git
synced 2025-04-21 00:40:01 +02:00
adds tests for methods exposed by DataService
This commit is contained in:
parent
9616c57c38
commit
a05b703bb8
@ -1,9 +1,14 @@
|
|||||||
from enum import Enum
|
from enum import Enum
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
import pydase
|
||||||
import pydase.units as u
|
import pydase.units as u
|
||||||
|
import pytest
|
||||||
from pydase import DataService
|
from pydase import DataService
|
||||||
from pydase.data_service.data_service_observer import DataServiceObserver
|
from pydase.data_service.data_service_observer import DataServiceObserver
|
||||||
from pydase.data_service.state_manager import StateManager
|
from pydase.data_service.state_manager import StateManager
|
||||||
|
from pydase.data_service.task_manager import TaskDefinitionError
|
||||||
|
from pydase.utils.serializer import FunctionDefinitionError, frontend
|
||||||
from pytest import LogCaptureFixture
|
from pytest import LogCaptureFixture
|
||||||
|
|
||||||
|
|
||||||
@ -114,3 +119,19 @@ def test_protected_and_private_attribute_warning(caplog: LogCaptureFixture) -> N
|
|||||||
"Class 'SubClass' does not inherit from DataService. This may lead to "
|
"Class 'SubClass' does not inherit from DataService. This may lead to "
|
||||||
"unexpected behaviour!"
|
"unexpected behaviour!"
|
||||||
) not in caplog.text
|
) not in caplog.text
|
||||||
|
|
||||||
|
|
||||||
|
def test_exposing_methods() -> None:
|
||||||
|
class ClassWithTask(pydase.DataService):
|
||||||
|
async def some_task(self, sleep_time: int) -> None:
|
||||||
|
pass
|
||||||
|
|
||||||
|
with pytest.raises(TaskDefinitionError):
|
||||||
|
ClassWithTask()
|
||||||
|
|
||||||
|
with pytest.raises(FunctionDefinitionError):
|
||||||
|
|
||||||
|
class ClassWithMethod(pydase.DataService):
|
||||||
|
@frontend
|
||||||
|
def some_method(self, *args: Any) -> str:
|
||||||
|
return "some method"
|
||||||
|
Loading…
x
Reference in New Issue
Block a user