tests/run/lift-and-unlift.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

import Function.unlift

object Test {
  def evens1(x: Int) = if (x % 2 == 0) Some(x) else None
  val evens2: PartialFunction[Int, Int] = {
    case x if x % 2 == 0  => x
  }

  def main(args: Array[String]): Unit = {
    val f1 = evens1 _
    val f2 = evens2.lift

    assert(1 to 10 forall (x => f1(x) == f2(x)))

    val f3 = unlift(f1)
    val f4 = unlift(f2)

    assert(1 to 10 forall { x =>
      if (!f3.isDefinedAt(x)) !f4.isDefinedAt(x)
      else f3(x) == f4(x)
    })

    assert(f1 eq f3.lift)
    assert(f4 eq unlift(f2))
    assert(f4 eq evens2)
  }
}