add tests

tests/unit_tests/zn/test_zn_options.py

@@ -6,6 +6,7 @@
 import znjson
 
 from zntrack import zn
+from zntrack.zn.dependencies import NodeAttribute, getdeps
 from zntrack.zn.split_option import combine_values, split_value
 
 
@@ -112,3 +113,19 @@ def test_split_value_path():
     new_path = combine_values(zntrack_data, params_data)
     # TODO change order to be consistent with split_values
     assert new_path == path
+
+
+class ExampleNode:
+    module = "module"
+    node_name = "node01"
+    affected_files = ["a", "b"]
+
+
+def test_getdeps():
+    assert getdeps(ExampleNode(), "my_attr") == NodeAttribute(
+        module="module",
+        cls="ExampleNode",
+        name="node01",
+        attribute="my_attr",
+        affected_files=["a", "b"],
+    )

zntrack/zn/dependencies.py

@@ -2,7 +2,7 @@
 
 import dataclasses
 import pathlib
-from typing import TYPE_CHECKING, Set
+from typing import TYPE_CHECKING, List
 
 from zntrack.utils import utils
 
@@ -16,7 +16,7 @@ class NodeAttribute:
     cls: str
     name: str
     attribute: str
-    affected_files: Set[pathlib.Path]
+    affected_files: List[pathlib.Path]
 
 
 def getdeps(node: Node, attribute: str) -> NodeAttribute: