From fe7486d527968879f413db9dbc5689a4aec5b82e Mon Sep 17 00:00:00 2001 From: jpalm3r Date: Wed, 18 Mar 2026 09:11:00 +0100 Subject: [PATCH] Import fix --- tests/test_network.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/tests/test_network.py b/tests/test_network.py index 92d0927fe..e1724b705 100644 --- a/tests/test_network.py +++ b/tests/test_network.py @@ -5,10 +5,12 @@ import xarray as xr import numpy as np import modelskill as ms -from modelskill.network import ( - Network, +from modelskill.model.network import ( NetworkModelResult, NodeModelResult, +) +from modelskill.network import ( + Network, BasicNode, BasicEdge, ) @@ -245,7 +247,8 @@ def test_node_attrs(self, sample_node_data): def test_multiple_nodes_returns_list_of_observations(self, multi_data): """Test that from_multiple returns a list of NodeObservation objects""" obs_list = NodeObservation.from_multiple( - data=multi_data, nodes={123: "station_0", 456: "station_1", 789: "station_2"} + data=multi_data, + nodes={123: "station_0", 456: "station_1", 789: "station_2"}, ) assert len(obs_list) == 3 @@ -253,7 +256,8 @@ def test_multiple_nodes_returns_list_of_observations(self, multi_data): def test_node_ids_are_assigned_correctly(self, multi_data): obs_list = NodeObservation.from_multiple( - data=multi_data, nodes={123: "station_0", 456: "station_1", 789: "station_2"} + data=multi_data, + nodes={123: "station_0", 456: "station_1", 789: "station_2"}, ) assert obs_list[0].node == 123 @@ -262,7 +266,8 @@ def test_node_ids_are_assigned_correctly(self, multi_data): def test_names_derived_from_column_names(self, multi_data): obs_list = NodeObservation.from_multiple( - data=multi_data, nodes={123: "station_0", 456: "station_1", 789: "station_2"} + data=multi_data, + nodes={123: "station_0", 456: "station_1", 789: "station_2"}, ) assert obs_list[0].name == "station_0"