range methods on SortedSet / SortedMap

build.sbt

@@ -11,7 +11,7 @@ unmanagedSourceDirectories in Compile += (
   else (sourceDirectory in Compile).value / "scala-2.11_2.12"
 )
 
-crossScalaVersions := Seq("2.12.5", "2.13.0-M4-pre-20d3c21", "2.11.12")
+crossScalaVersions := Seq("2.12.5", "2.13.0-pre-b11db01", "2.11.12")
 
 scalaVersion := crossScalaVersions.value.head
 

src/main/scala-2.11_2.12/collection/compat_impl/package.scala → src/main/scala-2.11_2.12/scala/collection/compat/CompatImpl.scala

@@ -1,10 +1,10 @@
-package scala.collection
+package scala.collection.compat
 
-import scala.collection.generic._
+import scala.collection.generic.CanBuildFrom
 import scala.collection.mutable.Builder
-import scala.reflect.ClassTag
+import scala.collection.{immutable, mutable}
 
-package object compat_impl {
+private[compat] object CompatImpl {
   def simpleCBF[A, C](f: => Builder[A, C]): CanBuildFrom[Any, A, C] = new CanBuildFrom[Any, A, C] {
     def apply(from: Any): Builder[A, C] = apply()
     def apply(): Builder[A, C] = f

src/main/scala-2.11_2.12/collection/compat/package.scala → src/main/scala-2.11_2.12/scala/collection/compat/package.scala

@@ -6,7 +6,7 @@ import scala.reflect.ClassTag
 
 /** The collection compatibility API */
 package object compat {
-  import scala.collection.compat_impl._
+  import CompatImpl._
 
   implicit def genericCompanionToCBF[A, CC[X] <: GenTraversable[X]](fact: GenericCompanion[CC]): CanBuildFrom[Any, A, CC[A]] =
     simpleCBF(fact.newBuilder[A])
@@ -48,6 +48,12 @@ package object compat {
     def lazyAppendAll(as: => TraversableOnce[A]): Stream[A] = stream.append(as)
   }
 
+  implicit class SortedExtensionMethods[K, T <: Sorted[K, T]](private val fact: Sorted[K, T]) {
+    def rangeFrom(from: K): T = fact.from(from)
+    def rangeTo(to: K): T = fact.to(to)
+    def rangeUntil(until: K): T = fact.until(until)
+  }
+
   // This really belongs into scala.collection but there's already a package object in scala-library so we can't add to it
   type IterableOnce[+X] = TraversableOnce[X]
 }

src/test/scala/collection/BuildFromTest.scala → src/test/scala/scala/collection/BuildFromTest.scala

@@ -2,9 +2,9 @@ package scala.collection
 
 import org.junit.Test
 
-import scala.collection.mutable.{ArrayBuffer, Builder, ListBuffer}
-import scala.collection.immutable.{HashMap, List, TreeMap, TreeSet}
 import scala.collection.compat._
+import scala.collection.immutable.{HashMap, List, TreeMap, TreeSet}
+import scala.collection.mutable.{ArrayBuffer, Builder, ListBuffer}
 
 // Tests copied from the 2.13 scala-library
 class BuildFromTest {

src/test/scala/collection/CollectionTest.scala → src/test/scala/scala/collection/CollectionTest.scala

@@ -1,12 +1,10 @@
-package collection
+package scala.collection
 
-import java.util
-
-import org.junit.Test
 import org.junit.Assert._
+import org.junit.Test
 
-import scala.collection.immutable.BitSet
 import scala.collection.compat._
+import scala.collection.immutable.{BitSet => BS} // importing BitSet doesn't work, scala.collection.BitSet shadows it
 
 class CollectionTest {
   @Test
@@ -20,9 +18,9 @@ class CollectionTest {
     val aT: Array[Int] = a
     assertEquals(Vector(1,2,3), a.toVector)
 
-    val b = xs.to(BitSet) // we can fake a type constructor for the 2 standard BitSet types
+    val b = xs.to(BS) // we can fake a type constructor for the 2 standard BitSet types
     val bT: BitSet = b
-    assertEquals(BitSet(1,2,3), b)
+    assertEquals(BS(1,2,3), b)
 
     val ys = List(1 -> "a", 2 -> "b")
     val m = ys.to(Map)
@@ -39,9 +37,9 @@ class CollectionTest {
     val vT: Vector[Int] = v
     assertEquals(Vector(1,2,3), v)
 
-    val b = BitSet.fromSpecific(xs)
-    val bT: BitSet = b
-    assertEquals(BitSet(1,2,3), b)
+    val b = BS.fromSpecific(xs)
+    val bT: BS = b
+    assertEquals(BS(1,2,3), b)
 
     val ys = List(1 -> "a", 2 -> "b")
     val m = Map.from(ys)

src/test/scala/collection/FactoryTest.scala → src/test/scala/scala/collection/FactoryTest.scala

@@ -1,8 +1,7 @@
-package collection
+package scala.collection
 
 import org.junit.{Assert, Test}
 
-import scala.collection.{BitSet, Factory, Iterable, immutable, mutable}
 import scala.collection.compat._
 
 class FactoryTest {

src/test/scala/collection/ImmutableArrayTest.scala → src/test/scala/scala/collection/ImmutableArrayTest.scala

@@ -1,4 +1,4 @@
-package collection
+package scala.collection
 
 import org.junit.{Assert, Test}
 

src/test/scala/collection/StreamTest.scala → src/test/scala/scala/collection/StreamTest.scala

@@ -1,4 +1,4 @@
-package collection
+package scala.collection
 
 import org.junit.Test
 

src/test/scala/scala/collection/generic/SortedTest.scala

@@ -0,0 +1,55 @@
+package scala.collection.generic
+
+import org.junit.Test
+import org.junit.Assert._
+
+import scala.collection.compat._
+import scala.collection.{immutable => i, mutable => m}
+
+class SortedTest {
+  @Test
+  def immutableRangeOps(): Unit = {
+    val st = i.SortedSet(-1, 2, 4, 3)
+    for (x <- List(0, 1, 4, -1, -4)) {
+      val s1 = st.rangeFrom(x)
+      assertEquals(s1: i.SortedSet[Int], st.from(x))
+      val s2 = st.rangeTo(x)
+      assertEquals(s2: i.SortedSet[Int], st.to(x))
+      val s3 = st.rangeUntil(x)
+      assertEquals(s3: i.SortedSet[Int], st.until(x))
+    }
+    val mp = i.SortedMap("" -> 0, "ds" -> -3, "-??" -> 13, "Df" -> 33, "d!" -> -32)
+    for (x <- List("", "-", "@", "aa", "D", "d")) {
+      val m1 = mp.rangeFrom(x)
+      assertEquals(m1: i.SortedMap[String, Int], mp.from(x))
+      val m2 = mp.rangeTo(x)
+      assertEquals(m2: i.SortedMap[String, Int], mp.to(x))
+      val m3 = mp.rangeUntil(x)
+      assertEquals(m3: i.SortedMap[String, Int], mp.until(x))
+    }
+  }
+
+  @Test
+  def mutableRangeOps(): Unit = {
+    val st = m.SortedSet(-1, 2, 4, 3)
+    for (x <- List(0, 1, 4, -1, -4)) {
+      val s1 = st.rangeFrom(x)
+      assertEquals(s1: m.SortedSet[Int], st.from(x))
+      val s2 = st.rangeTo(x)
+      assertEquals(s2: m.SortedSet[Int], st.to(x))
+      val s3 = st.rangeUntil(x)
+      assertEquals(s3: m.SortedSet[Int], st.until(x))
+    }
+    /* 2.11 doesn't have a mutable.SortedMap
+    val mp = m.SortedMap("" -> 0, "ds" -> -3, "-??" -> 13, "Df" -> 33, "d!" -> -32)
+    for (x <- List("", "-", "@", "aa", "D", "d")) {
+      val m1 = mp.rangeFrom(x)
+      assertEquals(m1: m.SortedMap[String, Int], mp.from(x))
+      val m2 = mp.rangeTo(x)
+      assertEquals(m2: m.SortedMap[String, Int], mp.to(x))
+      val m3 = mp.rangeUntil(x)
+      assertEquals(m3: m.SortedMap[String, Int], mp.until(x))
+    }
+    */
+  }
+}