Restore preamble id

Fix #46.
poliastro · Nov 7, 2019 · 21e74a2 · 21e74a2
1 parent 377e5fc
commit 21e74a2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/src/czml3/core.py b/src/czml3/core.py
@@ -12,14 +12,12 @@
 class Preamble(BaseCZMLObject):
     """The preamble packet."""
 
+    id = attr.ib(init=False, default="document")
+
     version = attr.ib(default=CZML_VERSION)
     name = attr.ib(default=None)
     clock = attr.ib(default=None)
 
-    @property
-    def id(self):
-        return "document"
-
 
 @attr.s(repr=False, frozen=True, kw_only=True)
 class Packet(BaseCZMLObject):

diff --git a/tests/test_core.py b/tests/test_core.py
@@ -0,0 +1,12 @@
+from czml3.core import Preamble
+
+
+def test_empty_preamble_has_id_and_version():
+    expected_result = """{
+    "id": "document",
+    "version": "1.0"
+}"""
+
+    packet = Preamble()
+
+    assert repr(packet) == expected_result