Gergő Jedlicska
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
import pytest
|
|
from specklepy.api.models import Stream
|
|
from specklepy.objects import Base
|
|
from specklepy.objects.encoding import ObjectArray
|
|
from specklepy.serialization.base_object_serializer import BaseObjectSerializer
|
|
from specklepy.transports.sqlite import SQLiteTransport
|
|
|
|
|
|
class TestObject:
|
|
@pytest.fixture(scope="module")
|
|
def stream(self, client):
|
|
stream = Stream(
|
|
name="a sample stream for testing",
|
|
description="a stream created for testing",
|
|
isPublic=True,
|
|
)
|
|
stream.id = client.stream.create(
|
|
stream.name, stream.description, stream.isPublic
|
|
)
|
|
return stream
|
|
|
|
def test_object_create(self, client, stream, base):
|
|
transport = SQLiteTransport()
|
|
s = BaseObjectSerializer(write_transports=[transport], read_transport=transport)
|
|
_, base_dict = s.traverse_base(base)
|
|
obj_id = client.object.create(stream_id=stream.id, objects=[base_dict])[0]
|
|
|
|
assert isinstance(obj_id, str)
|
|
assert base_dict["@detach"]["speckle_type"] == "reference"
|
|
assert obj_id == base.get_id(True)
|
|
|
|
def test_object_get(self, client, stream, base):
|
|
fetched_base = client.object.get(
|
|
stream_id=stream.id, object_id=base.get_id(True)
|
|
)
|
|
|
|
assert isinstance(fetched_base, Base)
|
|
assert fetched_base.name == base.name
|
|
assert isinstance(fetched_base.vertices, list)
|
|
# assert fetched_base["@detach"]["speckle_type"] == "reference"
|
|
|
|
def test_object_array_decoder(self):
|
|
array = ObjectArray()
|
|
array.data = [5, 1, 1, 1, 1, 1, 4, 1, 1, 1, 1, 3, 1, 1, 1, 2, 1, 1, 1, 1]
|
|
|
|
assert array.decode(decoder=sum) == [5, 4, 3, 2, 1]
|