Improve checks in test

NREL-Sienna · Feb 3, 2025 · f15586d · f15586d
1 parent 1d6a5ec
commit f15586d
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/test/test_system.jl b/test/test_system.jl
@@ -625,10 +625,22 @@ end
     @test orig != new_number
     set_bus_number!(sys, bus1, new_number)
     @test get_number(bus1) == new_number
+    bus_numbers = get_bus_numbers(sys)
+    @test new_number in bus_numbers
+    @test !(orig in bus_numbers)
+
+    # Ensure that the no-op case works.
+    set_bus_number!(sys, bus1, new_number)
+    @test get_number(bus1) == new_number
+    @test new_number in get_bus_numbers(sys)
+
+    # Ensure that duplicate numbers are blocked.
     @test_throws ArgumentError set_bus_number!(sys, bus1, get_number(bus2))
-    @test !(orig in get_bus_numbers(sys))
+
+    # Esnure that you can't change an unattached bus.
     remove_component!(sys, bus1)
     @test_throws ArgumentError set_bus_number!(sys, bus1, new_number + 1)
+
     # Ensure that this is exported. This can be deleted in PSY5.
     set_number!(bus1, new_number + 2)
     @test get_number(bus1) == new_number + 2