move CoordinatedNodeArray

Signed-off-by: Thijs Baaijen <[email protected]>

Author: Thijss

tests/integration/visualizer_tests.py

@@ -4,18 +4,10 @@
 
 from dataclasses import dataclass
 
-import numpy as np
-from numpy.typing import NDArray
-
 from power_grid_model_ds import Grid
 from power_grid_model_ds._core.visualizer.main import visualize
-from power_grid_model_ds.arrays import NodeArray
 from power_grid_model_ds.generators import RadialGridGenerator
-
-
-class CoordinatedNodeArray(NodeArray):
-    x: NDArray[np.float64]
-    y: NDArray[np.float64]
+from tests.unit.visualizer.test_parsers import CoordinatedNodeArray
 
 
 @dataclass

tests/unit/visualizer/test_parsers.py

@@ -2,9 +2,17 @@
 #
 # SPDX-License-Identifier: MPL-2.0
 
+
+import numpy as np
+from numpy.typing import NDArray
+
 from power_grid_model_ds._core.model.arrays import LineArray, NodeArray
 from power_grid_model_ds._core.visualizer.parsers import parse_branch_array, parse_node_array
-from tests.integration.visualizer_tests import CoordinatedNodeArray
+
+
+class CoordinatedNodeArray(NodeArray):
+    x: NDArray[np.float64]
+    y: NDArray[np.float64]
 
 
 class TestParseNodeArray: