core: fix distance_range_map.to_range_map

Author: Castavo

core/src/main/kotlin/fr/sncf/osrd/standalone_sim/StandaloneSimulation.kt

@@ -422,7 +422,7 @@ private fun <T> DistanceRangeMap<T>.toRangeMap(): RangeMap<Double, T> {
     val res = ImmutableRangeMap.builder<Double, T>()
     for (entry in this) {
         if (entry.value != null)
-            res.put(Range.closed(entry.lower.meters, entry.upper.meters), entry.value!!)
+            res.put(Range.closedOpen(entry.lower.meters, entry.upper.meters), entry.value!!)
     }
     return res.build()
 }

core/src/test/kotlin/fr/sncf/osrd/standalone_sim/StandaloneSimulationTest.kt

@@ -18,14 +18,8 @@ import fr.sncf.osrd.railjson.schema.rollingstock.Comfort
 import fr.sncf.osrd.railjson.schema.schedule.RJSAllowanceDistribution
 import fr.sncf.osrd.sim_infra.api.makePathProperties
 import fr.sncf.osrd.train.TestTrains
-import fr.sncf.osrd.utils.Helpers
-import fr.sncf.osrd.utils.distanceRangeMapOf
-import fr.sncf.osrd.utils.pathFromRoutes
-import fr.sncf.osrd.utils.toIdxList
-import fr.sncf.osrd.utils.units.Distance
-import fr.sncf.osrd.utils.units.Offset
-import fr.sncf.osrd.utils.units.TimeDelta
-import fr.sncf.osrd.utils.units.seconds
+import fr.sncf.osrd.utils.*
+import fr.sncf.osrd.utils.units.*
 import java.util.stream.Stream
 import kotlin.test.assertEquals
 import org.junit.jupiter.api.Test
@@ -39,10 +33,10 @@ class StandaloneSimulationTest {
     private val infra = Helpers.tinyInfra
     private val routes =
         listOf(
-                "rt.buffer_stop_c->tde.track-bar",
-                "rt.tde.track-bar->tde.switch_foo-track",
-                "rt.tde.switch_foo-track->buffer_stop_a"
-            )
+            "rt.buffer_stop_c->tde.track-bar",
+            "rt.tde.track-bar->tde.switch_foo-track",
+            "rt.tde.switch_foo-track->buffer_stop_a"
+        )
             .map { infra.rawInfra.getRouteFromName(it) }
 
     private val chunkPath = pathFromRoutes(infra.rawInfra, routes)
@@ -99,6 +93,7 @@ class StandaloneSimulationTest {
         val startSpeed: Double = 0.0,
         val margins: RangeValues<MarginValue> = RangeValues(),
         val pathLength: Distance,
+        val powerRestrictions: DistanceRangeMap<String> = distanceRangeMapOf()
     )
 
     /**
@@ -163,27 +158,43 @@ class StandaloneSimulationTest {
                 )
             )
 
+        // Power restriction values
+        val powerRestrictionRangeMaps: List<DistanceRangeMap<String>> = listOf(
+            distanceRangeMapOf(),
+            distanceRangeMapOf(
+                listOf(
+                    DistanceRangeMap.RangeMapEntry(0.meters, pathLength / 3.0, "Restrict1"),
+                    DistanceRangeMap.RangeMapEntry(pathLength / 3.0, pathLength * 2.0 / 3.0, "Restrict2"),
+                    DistanceRangeMap.RangeMapEntry(pathLength * 2.0 / 3.0, pathLength, "Restrict1")
+                )
+            )
+        )
+
         // List all possible combinations
         val res = mutableListOf<TestCase>()
         for (schedule in schedules) {
             for (margin in margins) {
                 for (startSpeed in listOf(0.0, 15.0)) {
                     for (distribution in RJSAllowanceDistribution.entries) {
-                        res.add(
-                            TestCase(
-                                schedule = schedule,
-                                margins = margin,
-                                startSpeed = startSpeed,
-                                allowanceDistribution = distribution,
-                                pathLength = pathLength
+                        for (powerRestrictions in powerRestrictionRangeMaps) {
+                            res.add(
+                                TestCase(
+                                    schedule = schedule,
+                                    margins = margin,
+                                    startSpeed = startSpeed,
+                                    allowanceDistribution = distribution,
+                                    pathLength = pathLength,
+                                    powerRestrictions = powerRestrictions
+                                )
                             )
-                        )
+                        }
                     }
                 }
             }
         }
         return res.map { Arguments.of(it) }.stream()
     }
+
     /** Parametrized test, checks the interactions between margins and scheduled points */
     @ParameterizedTest
     @MethodSource("generateTestCases")
@@ -199,7 +210,7 @@ class StandaloneSimulationTest {
                 Comfort.STANDARD,
                 testCase.allowanceDistribution,
                 null,
-                distanceRangeMapOf(),
+                testCase.powerRestrictions,
                 false,
                 2.0,
                 testCase.schedule,