adjust unit tests for NodeDefinition

sevmag · sevmag · commit 4384ff5f7b0b · 2025-03-18T14:53:54.000+01:00
diff --git a/tests/models/test_node_definition.py b/tests/models/test_node_definition.py
@@ -38,7 +38,8 @@ def test_percentile_cluster() -> None:
     )
 
     # Apply node definition to torch tensor with raw pulses
-    graph, new_features = node_definition(tensor)
+    graph = node_definition(tensor)
+    new_features = node_definition._output_feature_names
     x_tilde = graph.x.numpy()
 
     # Calculate percentiles "the normal way" and compare that output of

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,8 @@ def test_percentile_cluster() -> None:`
`38`	`38`	`)`
`39`	`39`
`40`	`40`	`# Apply node definition to torch tensor with raw pulses`
`41`		`- graph, new_features = node_definition(tensor)`
	`41`	`+ graph = node_definition(tensor)`
	`42`	`+ new_features = node_definition._output_feature_names`
`42`	`43`	`x_tilde = graph.x.numpy()`
`43`	`44`
`44`	`45`	`# Calculate percentiles "the normal way" and compare that output of`