1 files changed, 129 insertions, 0 deletions
diff --git a/src/partest-extras/scala/org/scalacheck/util/Pretty.scala b/src/partest-extras/scala/org/scalacheck/util/Pretty.scala
new file mode 100644
index 0000000000..13a1b44b51
--- /dev/null
+++ b/src/partest-extras/scala/org/scalacheck/util/Pretty.scala
@@ -0,0 +1,129 @@
+/*-------------------------------------------------------------------------*\
+**  ScalaCheck                                                             **
+**  Copyright (c) 2007-2014 Rickard Nilsson. All rights reserved.          **
+**  http://www.scalacheck.org                                              **
+**                                                                         **
+**  This software is released under the terms of the Revised BSD License.  **
+**  There is NO WARRANTY. See the file LICENSE for the full text.          **
+\*------------------------------------------------------------------------ */
+
+package org.scalacheck.util
+
+import org.scalacheck.Prop.Arg
+import org.scalacheck.Test
+
+import math.round
+
+
+sealed trait Pretty {
+  def apply(prms: Pretty.Params): String
+
+  def map(f: String => String) = Pretty(prms => f(Pretty.this(prms)))
+
+  def flatMap(f: String => Pretty) = Pretty(prms => f(Pretty.this(prms))(prms))
+}
+
+object Pretty {
+
+  case class Params(verbosity: Int)
+
+  val defaultParams = Params(0)
+
+  def apply(f: Params => String) = new Pretty { def apply(p: Params) = f(p) }
+
+  def pretty[T <% Pretty](t: T, prms: Params): String = t(prms)
+
+  def pretty[T <% Pretty](t: T): String = t(defaultParams)
+
+  implicit def strBreak(s1: String) = new {
+    def /(s2: String) = if(s2 == "") s1 else s1+"\n"+s2
+  }
+
+  def pad(s: String, c: Char, length: Int) =
+    if(s.length >= length) s
+    else s + List.fill(length-s.length)(c).mkString
+
+  def break(s: String, lead: String, length: Int): String =
+    if(s.length <= length) s
+    else s.substring(0, length) / break(lead+s.substring(length), lead, length)
+
+  def format(s: String, lead: String, trail: String, width: Int) =
+    s.lines.map(l => break(lead+l+trail, "  ", width)).mkString("\n")
+
+  implicit def prettyAny(t: Any) = Pretty { p => t.toString }
+
+  implicit def prettyString(t: String) = Pretty { p => "\""++t++"\"" }
+
+  implicit def prettyList(l: List[Any]) = Pretty { p =>
+    l.map("\""+_+"\"").mkString("List(", ", ", ")")
+  }
+
+  implicit def prettyThrowable(e: Throwable) = Pretty { prms =>
+    val strs = e.getStackTrace.map { st =>
+      import st._
+      getClassName+"."+getMethodName + "("+getFileName+":"+getLineNumber+")"
+    }
+
+    val strs2 =
+      if(prms.verbosity <= 0) Array[String]()
+      else if(prms.verbosity <= 1) strs.take(5)
+      else strs
+
+    e.getClass.getName + ": " + e.getMessage / strs2.mkString("\n")
+  }
+
+  def prettyArgs(args: Seq[Arg[Any]]): Pretty = Pretty { prms =>
+    if(args.isEmpty) "" else {
+      for((a,i) <- args.zipWithIndex) yield {
+        val l = "> "+(if(a.label == "") "ARG_"+i else a.label)
+        val s =
+          if(a.shrinks == 0) ""
+          else "\n"+l+"_ORIGINAL: "+a.prettyOrigArg(prms)
+        l+": "+a.prettyArg(prms)+""+s
+      }
+    }.mkString("\n")
+  }
+
+  implicit def prettyFreqMap(fm: FreqMap[Set[Any]]) = Pretty { prms =>
+    if(fm.total == 0) ""
+    else {
+      "> Collected test data: " / {
+        for {
+          (xs,r) <- fm.getRatios
+          ys = xs - (())
+          if !ys.isEmpty
+        } yield round(r*100)+"% " + ys.mkString(", ")
+      }.mkString("\n")
+    }
+  }
+
+  implicit def prettyTestRes(res: Test.Result) = Pretty { prms =>
+    def labels(ls: collection.immutable.Set[String]) =
+      if(ls.isEmpty) ""
+      else "> Labels of failing property: " / ls.mkString("\n")
+    val s = res.status match {
+      case Test.Proved(args) => "OK, proved property."/prettyArgs(args)(prms)
+      case Test.Passed => "OK, passed "+res.succeeded+" tests."
+      case Test.Failed(args, l) =>
+        "Falsified after "+res.succeeded+" passed tests."/labels(l)/prettyArgs(args)(prms)
+      case Test.Exhausted =>
+        "Gave up after only "+res.succeeded+" passed tests. " +
+        res.discarded+" tests were discarded."
+      case Test.PropException(args,e,l) =>
+        "Exception raised on property evaluation."/labels(l)/prettyArgs(args)(prms)/
+        "> Exception: "+pretty(e,prms)
+      case Test.GenException(e) =>
+        "Exception raised on argument generation."/
+        "> Exception: "+pretty(e,prms)
+    }
+    val t = if(prms.verbosity <= 1) "" else "Elapsed time: "+prettyTime(res.time)
+    s/t/pretty(res.freqMap,prms)
+  }
+
+  def prettyTime(millis: Long): String = {
+    val min = millis/(60*1000)
+    val sec = (millis-(60*1000*min)) / 1000d
+    if(min <= 0) "%.3f sec ".format(sec)
+    else "%d min %.3f sec ".format(min, sec)
+  }
+}