test protected node attributes

zincware · Aug 29, 2024 · a44e427 · a44e427
1 parent 9fc2b31
commit a44e427
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_protected.py b/tests/test_protected.py
@@ -0,0 +1,15 @@
+import znflow
+
+
+class MyNode(znflow.Node):
+    _protected_ = znflow.Node._protected_ + ["a"]
+
+    a: int = 42
+    b: int = 42
+
+
+def test_protected():
+    with znflow.DiGraph():
+        node = MyNode()
+        assert node.a == 42
+        assert isinstance(node.b, znflow.Connection)