mirror of
https://github.com/tiqi-group/pydase.git
synced 2025-04-20 00:10:03 +02:00
102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
from pytest import CaptureFixture
|
|
|
|
from pydase import DataService
|
|
|
|
|
|
def test_class_list_attribute(capsys: CaptureFixture) -> None:
|
|
class ServiceClass(DataService):
|
|
attr = [0, 1]
|
|
|
|
service_instance = ServiceClass()
|
|
|
|
service_instance.attr[0] = 1337
|
|
captured = capsys.readouterr()
|
|
assert captured.out == "ServiceClass.attr[0] = 1337\n"
|
|
|
|
|
|
def test_instance_list_attribute(capsys: CaptureFixture) -> None:
|
|
class SubClass(DataService):
|
|
name = "SubClass"
|
|
|
|
class ServiceClass(DataService):
|
|
def __init__(self) -> None:
|
|
self.attr = [0, SubClass()]
|
|
super().__init__()
|
|
|
|
service_instance = ServiceClass()
|
|
|
|
_ = capsys.readouterr()
|
|
|
|
service_instance.attr[0] = "Hello"
|
|
captured = capsys.readouterr()
|
|
assert captured.out == "ServiceClass.attr[0] = Hello\n"
|
|
|
|
service_instance.attr[1] = SubClass()
|
|
captured = capsys.readouterr()
|
|
assert (
|
|
captured.out.strip()
|
|
== "ServiceClass.attr[1] = {'name': {'type': 'str', 'value': 'SubClass',"
|
|
" 'readonly': False, 'doc': None}}"
|
|
)
|
|
|
|
|
|
def test_reused_instance_list_attribute(capsys: CaptureFixture) -> None:
|
|
some_list = [0, 1, 2]
|
|
|
|
class ServiceClass(DataService):
|
|
def __init__(self) -> None:
|
|
self.attr = some_list
|
|
self.attr_2 = some_list
|
|
self.attr_3 = [0, 1, 2]
|
|
super().__init__()
|
|
|
|
service_instance = ServiceClass()
|
|
|
|
service_instance.attr[0] = 20
|
|
captured = capsys.readouterr()
|
|
|
|
assert service_instance.attr == service_instance.attr_2
|
|
assert service_instance.attr != service_instance.attr_3
|
|
expected_output = sorted(
|
|
[
|
|
"ServiceClass.attr[0] = 20",
|
|
"ServiceClass.attr_2[0] = 20",
|
|
]
|
|
)
|
|
actual_output = sorted(captured.out.strip().split("\n"))
|
|
assert actual_output == expected_output
|
|
|
|
|
|
def test_nested_reused_instance_list_attribute(capsys: CaptureFixture) -> None:
|
|
some_list = [0, 1, 2]
|
|
|
|
class SubClass(DataService):
|
|
attr_list = some_list
|
|
|
|
def __init__(self) -> None:
|
|
self.attr_list_2 = some_list
|
|
super().__init__()
|
|
|
|
class ServiceClass(DataService):
|
|
def __init__(self) -> None:
|
|
self.attr = some_list
|
|
self.subclass = SubClass()
|
|
super().__init__()
|
|
|
|
service_instance = ServiceClass()
|
|
|
|
_ = capsys.readouterr()
|
|
service_instance.attr[0] = 20
|
|
captured = capsys.readouterr()
|
|
|
|
assert service_instance.attr == service_instance.subclass.attr_list
|
|
expected_output = sorted(
|
|
[
|
|
"ServiceClass.subclass.attr_list_2[0] = 20",
|
|
"ServiceClass.subclass.attr_list[0] = 20",
|
|
"ServiceClass.attr[0] = 20",
|
|
]
|
|
)
|
|
actual_output = sorted(captured.out.strip().split("\n"))
|
|
assert actual_output == expected_output
|