add test for scene serialisation

Author: Licini

tests/compas/scene/test_scene_serialisation.py

@@ -0,0 +1,116 @@
+import pytest  # noqa: F401
+
+from compas.data import Data
+from compas.scene import Scene
+from compas.geometry import Box
+from compas.geometry import Capsule
+from compas.geometry import Circle
+from compas.geometry import Cone
+from compas.geometry import Cylinder
+from compas.geometry import Ellipse
+from compas.geometry import Frame
+from compas.geometry import Line
+from compas.geometry import Point
+from compas.geometry import Polygon
+from compas.geometry import Polyhedron
+from compas.geometry import Polyline
+from compas.geometry import Sphere
+from compas.geometry import Torus
+from compas.geometry import Vector
+from compas.geometry import Plane
+from compas.datastructures import Mesh
+from compas.datastructures import Graph
+from compas.datastructures import VolMesh
+
+
+@pytest.fixture
+def items():
+    box = Box.from_width_height_depth(1, 1, 1)
+    capsule = Capsule(0.5, 1, Frame.worldXY())
+    circle = Circle(1, Frame.worldXY())
+    cone = Cone(1, 1, Frame.worldXY())
+    cylinder = Cylinder(1, 1, Frame.worldXY())
+    line = Line(Point(0, 0, 0), Point(1, 1, 1))
+    point = Point(0, 0, 0)
+    polygon = Polygon.from_sides_and_radius_xy(5, 1)
+    polyhedron = Polyhedron.from_platonicsolid(4)
+    polyline = Polyline([[0, 0, 0], [1, 0, 0], [1, 0, 1]])
+    sphere = Sphere(1)
+    torus = Torus(1, 0.3, Frame.worldXY())
+    vector = Vector(0, 0, 1)
+    ellipse = Ellipse(1, 0.5, Frame.worldXY())
+    frame = Frame.worldXY()
+    plane = Plane(Point(0, 0, 0), Vector(0, 0, 1))
+    mesh = Mesh.from_polyhedron(8)
+    graph = Graph.from_nodes_and_edges([(0, 0, 0), (0, -1.5, 0), (-1, 1, 0), (1, 1, 0)], [(0, 1), (0, 2), (0, 3)])
+    volmesh = VolMesh.from_meshgrid(1, 1, 1, 2, 2, 2)
+
+    return [
+        box,
+        capsule,
+        circle,
+        cone,
+        cylinder,
+        line,
+        point,
+        polygon,
+        polyhedron,
+        polyline,
+        sphere,
+        torus,
+        vector,
+        ellipse,
+        frame,
+        plane,
+        mesh,
+        graph,
+        volmesh,
+    ]
+
+
+def assert_is_data_equal(obj1, obj2, path=""):
+
+    if type(obj1) is not type(obj2):
+        print(f"Type mismatch: {type(obj1)} != {type(obj2)} for {path}:{obj1} and {path}:{obj2}")
+        return False
+
+    if isinstance(obj1, (list, tuple)):
+        if len(obj1) != len(obj2):
+            print(f"Length mismatch: {len(obj1)} != {len(obj2)} for {path} and {path}")
+            return False
+
+        for i, (item1, item2) in enumerate(zip(obj1, obj2)):
+            if not assert_is_data_equal(item1, item2, path=f"{path}[{i}]"):
+                return False
+
+        return True
+
+    elif isinstance(obj1, dict):
+        if len(obj1) != len(obj2):
+            print(f"Length mismatch: {len(obj1)} != {len(obj2)} for {path} and {path}")
+            return False
+
+        for key in obj1:
+            if not assert_is_data_equal(obj1[key], obj2[key], path=f'{path}["{key}"]'):
+                return False
+
+        return True
+
+    elif isinstance(obj1, Data):
+        return assert_is_data_equal(obj1.__data__, obj2.__data__, path=f"{path}.__data__")
+
+    else:
+        if obj1 != obj2:
+            print(f"Value mismatch: {obj1} != {obj2} for {path}:{obj1} and {path}:{obj2}")
+            return False
+        else:
+            return True
+
+
+def test_scene_serialisation(items):
+    scene1 = Scene()
+    for item in items:
+        scene1.add(item)
+
+    scene2 = Scene.from_jsonstring(scene1.to_jsonstring())
+    assert assert_is_data_equal(scene1, scene2)