summaryrefslogtreecommitdiff
path: root/test/files/scalacheck/parallel-collections/ParallelHashSetCheck.scala
blob: 973a5cdf4bdd7b87cf4adfc82ab8dee435002009 (plain) (blame)
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package scala.collection.parallel
package mutable



import org.scalacheck._
import org.scalacheck.Gen
import org.scalacheck.Gen._
import org.scalacheck.Prop._
import org.scalacheck.Properties
import org.scalacheck.Arbitrary._

import scala.collection._
import scala.collection.parallel.ops._


abstract class ParallelHashSetCheck[T](tp: String) extends ParallelSetCheck[T]("mutable.ParHashSet[" + tp + "]") {
  ForkJoinTasks.defaultForkJoinPool.setMaximumPoolSize(Runtime.getRuntime.availableProcessors * 2)
  ForkJoinTasks.defaultForkJoinPool.setParallelism(Runtime.getRuntime.availableProcessors * 2)

  type CollType = ParHashSet[T]

  def isCheckingViews = false

  def hasStrictOrder = false

  def ofSize(vals: Seq[Gen[T]], sz: Int) = {
    val hm = new mutable.HashSet[T]
    val gen = vals(rnd.nextInt(vals.size))
    for (i <- 0 until sz) hm += sample(gen)
    hm
  }

  def fromTraversable(t: Traversable[T]) = {
    val phm = new ParHashSet[T]
    var i = 0
    for (kv <- t.toList) {
      phm += kv
      i += 1
    }
    phm
  }

}


object IntParallelHashSetCheck extends ParallelHashSetCheck[Int]("Int")
with IntOperators
with IntValues
{
  override def printDataStructureDebugInfo(ds: AnyRef) = ds match {
    case pm: ParHashSet[t] =>
      println("Mutable parallel hash set")
    case _ =>
      println("could not match data structure type: " + ds.getClass)
  }

  override def checkDataStructureInvariants(orig: Traversable[Int], ds: AnyRef) = ds match {
    case pm: ParHashSet[t] =>
      // for an example of how not to write code proceed below
      val invs = pm.brokenInvariants

      val containsall = (for (elem <- orig) yield {
        if (pm.asInstanceOf[ParHashSet[Int]](elem) == true) true
        else {
          println("Does not contain original element: " + elem)
          println(pm.hashTableContents.table.find(_ == elem))
          println(pm.hashTableContents.table.indexOf(elem))
          false
        }
      }).foldLeft(true)(_ && _)


      if (invs.isEmpty) {
        if (!containsall) println(pm.debugInformation)
        containsall
      } else {
        println("Invariants broken:\n" + invs.mkString("\n"))
        false
      }
    case _ => true
  }

}