//############################################################################
// Lists
//############################################################################
// $Id$
//############################################################################
object Test {
val xs1 = List(1, 2, 3)
val xs2 = List('a', 'b')
val xs3 = List(List(1, 2), List(4, 5))
val xs4 = List(2, 4, 6, 8)
val xs5 = List(List(3, 4), List(3), List(4, 5))
def check_count: Int = {
val n1 = xs1 count { e => e % 2 != 0 }
val n2 = xs4 count { e => e < 5 }
n1 + n2
}
def check_diff: Int = {
val ys1 = xs1 diff xs4
val ys2 = xs3 diff xs5
ys1.length + ys2.length
}
def check_exists: Boolean = {
val b1 = xs1 exists { e => e % 2 == 0 }
val b2 = xs4 exists { e => e == 5 }
b1 & b2
}
def check_filter: Int = {
val ys1 = xs1 filter { e => e % 2 == 0 }
val ys2 = xs4 filter { e => e < 5 }
ys1.length + ys2.length
}
def check_foldLeft: Int = {
val n1 = xs1.foldLeft(0)((e1, e2) => e1 + e2)
val ys1 = xs4.foldLeft(List[Int]())((e1, e2) => e2 :: e1)
n1 + ys1.length
}
def check_forall: Boolean = {
val b1 = xs1 forall { e => e < 10}
val b2 = xs4 forall { e => e % 2 == 0 }
b1 & b2
}
def check_intersect: Int = {
val ys1 = xs1 intersect xs4
val ys2 = xs3 intersect xs5
ys1.length + ys2.length
}
def check_remove: Int = {
val ys1 = xs1 remove { e => e % 2 != 0 }
val ys2 = xs4 remove { e => e < 5 }
ys1.length + ys2.length
}
def check_union: Int = {
val ys1 = xs1 union xs4;
val ys2 = xs3 union xs5;
ys1.length + ys2.length
}
def check_zip: Int = {
val ys1 = xs1 zip xs2;
val ys2 = xs1 zip xs3;
ys1.length + ys2.length
}
def check_zipAll: Int = {
val ys1 = xs1.zipAll(xs2, 0, '_')
val ys2 = xs2.zipAll(xs1, '_', 0)
val ys3 = xs1.zipAll(xs3, 0, List(-1))
ys1.length + ys2.length + ys3.length
}
def check_success[A](name: String, closure: => A, expected: A) {
print("test " + name)
try {
val actual: A = closure
if (actual == expected)
print(" was successful")
else
print(" failed: expected "+ expected +", found "+ actual)
}
catch {
case exception: Throwable =>
print(" raised exception " + exception)
}
println
}
def main(args: Array[String]) {
check_success("check_count", check_count, 4)
check_success("check_diff", check_diff, 3)
check_success("check_exists", check_exists, false)
check_success("check_filter", check_filter, 3)
check_success("check_foldLeft", check_foldLeft, 10)
check_success("check_forall", check_forall, true)
check_success("check_intersect", check_intersect, 2)
check_success("check_remove", check_remove, 3)
check_success("check_union", check_union, 10)
check_success("check_zip", check_zip, 4)
check_success("check_zipAll", check_zipAll, 9)
println
}
}
//############################################################################
