add groovy bindings tests

Signed-off-by: RALAMBOTIANA MIORA <miora.ralambotiana@rte-france.com>
powsybl · Mar 5, 2020 · 8610ee5 · 8610ee5
1 parent b54287d
commit 8610ee5
Showing 1 changed file with 86 additions and 6 deletions.
diff --git a/...g/src/test/groovy/com/powsybl/iidm/network/scripting/ShuntCompensatorExtensionTest.groovy b/...g/src/test/groovy/com/powsybl/iidm/network/scripting/ShuntCompensatorExtensionTest.groovy
@@ -6,14 +6,17 @@
  */
 package com.powsybl.iidm.network.scripting
 
-import com.powsybl.iidm.network.Bus
+import com.powsybl.commons.PowsyblException
 import com.powsybl.iidm.network.Country
 import com.powsybl.iidm.network.Network
 import com.powsybl.iidm.network.ShuntCompensator
+import com.powsybl.iidm.network.ShuntCompensatorAdder
 import com.powsybl.iidm.network.Substation
 import com.powsybl.iidm.network.TopologyKind
 import com.powsybl.iidm.network.VoltageLevel
+import org.junit.Rule
 import org.junit.Test
+import org.junit.rules.ExpectedException
 
 import static org.junit.Assert.assertEquals
 
@@ -23,9 +26,12 @@ import static org.junit.Assert.assertEquals
  */
 class ShuntCompensatorExtensionTest {
 
+    @Rule
+    public ExpectedException thrown = ExpectedException.none()
+
     @Test
-    void test() {
-        ShuntCompensator shunt = createShuntCompensator()
+    void linearShuntTest() {
+        ShuntCompensator shunt = createLinearShuntCompensator()
         assertEquals(10, shunt.maximumSectionCount)
         shunt.maximumSectionCount = 11
         assertEquals(11, shunt.maximumSectionCount)
@@ -35,21 +41,52 @@ class ShuntCompensatorExtensionTest {
         assertEquals(44.0, shunt.maximumB, 0.0)
     }
 
-    static ShuntCompensator createShuntCompensator() {
+    @Test
+    void nonLinearShuntTest() {
+        ShuntCompensator shunt = createNonLinearShuntCompensator()
+        assertEquals(2, shunt.maximumSectionCount)
+        assertEquals(6.0, shunt.maximumB, 0.0f)
+    }
+
+    @Test
+    void failGetbPerSectionTest() {
+        thrown.expect(PowsyblException.class)
+        thrown.expectMessage("shunt model is not linear")
+        ShuntCompensator shunt = createNonLinearShuntCompensator()
+        double bPerSection = shunt.bPerSection
+    }
+
+    @Test
+    void failSetbPerSectionTest() {
+        thrown.expect(PowsyblException.class)
+        thrown.expectMessage("shunt model is not linear")
+        ShuntCompensator shunt = createNonLinearShuntCompensator()
+        shunt.bPerSection = 4.0
+    }
+
+    @Test
+    void failSetMaximumSectionCountTest() {
+        thrown.expect(PowsyblException.class)
+        thrown.expectMessage("shunt model is not linear")
+        ShuntCompensator shunt = createNonLinearShuntCompensator()
+        shunt.maximumSectionCount = 11
+    }
+
+    static ShuntCompensator createLinearShuntCompensator() {
         Network network = Network.create("test", "test")
         Substation substation = network.newSubstation()
                 .setCountry(Country.AF)
                 .setTso("tso")
                 .setName("sub")
                 .setId("subId")
-                .add();
+                .add()
         VoltageLevel voltageLevel = substation.newVoltageLevel()
                 .setTopologyKind(TopologyKind.BUS_BREAKER)
                 .setId("bbVL")
                 .setName("bbVL_name")
                 .setNominalV(200.0f)
                 .add()
-        Bus bus = voltageLevel.getBusBreakerView()
+        voltageLevel.getBusBreakerView()
                 .newBus()
                 .setName("Bus1")
                 .setId("Bus1")
@@ -64,4 +101,47 @@ class ShuntCompensatorExtensionTest {
                     .add()
                 .add()
     }
+
+    static ShuntCompensator createNonLinearShuntCompensator() {
+        Network network = Network.create("test", "test")
+        Substation substation = network.newSubstation()
+                .setCountry(Country.AF)
+                .setTso("tso")
+                .setName("sub")
+                .setId("subId")
+                .add()
+        VoltageLevel voltageLevel = substation.newVoltageLevel()
+                .setTopologyKind(TopologyKind.BUS_BREAKER)
+                .setId("bbVL")
+                .setName("bbVL_name")
+                .setNominalV(200.0f)
+                .add()
+        voltageLevel.getBusBreakerView()
+                .newBus()
+                .setName("Bus1")
+                .setId("Bus1")
+                .add()
+        ShuntCompensatorAdder adder = voltageLevel.newShuntCompensator()
+                .setId("SHUNT")
+                .setBus("Bus1")
+                .setConnectableBus("Bus1")
+                .setCurrentSectionCount(1)
+        adder.newNonLinearModel()
+                .beginSection()
+                    .setSectionNum(0)
+                    .setB(0.0)
+                    .setG(0.0)
+                .endSection()
+                .beginSection()
+                    .setSectionNum(1)
+                    .setB(5.0)
+                    .setG(2.0)
+                .endSection()
+                .beginSection()
+                    .setSectionNum(2)
+                    .setB(6.0)
+                .endSection()
+                .add()
+        adder.add()
+    }
 }