Test fix

Deltares · May 21, 2024 · 4041581 · 4041581
1 parent 99d71a2
commit 4041581
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/core/test/allocation_test.jl b/core/test/allocation_test.jl
@@ -177,9 +177,10 @@ end
     # Subnetworks interpreted as user_demands require variables and constraints to
     # support absolute value expressions in the objective function
     problem = allocation_model_main_network.problem
-    @test problem[:F_abs_user_demand].axes[1] == NodeID.(:Pump, [11, 24, 38])
-    @test problem[:abs_positive_user_demand].axes[1] == NodeID.(:Pump, [11, 24, 38])
-    @test problem[:abs_negative_user_demand].axes[1] == NodeID.(:Pump, [11, 24, 38])
+    node_ids = [NodeID(:UserDemand, 60), NodeID.(:Pump, [11, 24, 38])...]
+    @test problem[:F_abs_user_demand].axes[1] == node_ids
+    @test problem[:abs_positive_user_demand].axes[1] == node_ids
+    @test problem[:abs_negative_user_demand].axes[1] == node_ids
 
     # In each subnetwork, the connection from the main network to the subnetwork is
     # interpreted as a source