summaryrefslogtreecommitdiff
path: root/pbt/ts/src/property.ts
blob: a8c59ba61e3b15e3086562f8da45c7e90ba91f0f (plain)
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
import Prando from "prando";

type Result = 'OK' | 'Falsified' | 'Exception';

type ShrinkResult<A> = { counterexample: A, shrinks: number };

type TestResult<A> = {
  result: Result,
  counterexample: ShrinkResult<A> | null
};

type Predicate<A> = (a: A) => boolean;

export type Gen<A> = (rng: Prando) => ((size: number) => A);

type Shrinker<A> = (a: A) => A[];

type Property<A> = (rng: Prando,
                    size: number) =>  TestResult<A>;

const MAX_SUCCESS = 100;

function findMinimalCounterExample<A>(x: A,
  predicate: Predicate<A>,
  shrinker: Shrinker<A>,
  depth: number = 0): ShrinkResult<A> {
  let xs = shrinker(x);
  let counterexample = x;
  let shrinks = depth;
  for (let y of xs) {
    if (!predicate(y)) {
      let shrink = findMinimalCounterExample(y, predicate, shrinker, depth + 1);
      if (shrink.shrinks > depth) {
        counterexample = shrink.counterexample;
        shrinks = shrink.shrinks;
      }
    }
  }
  return { counterexample, shrinks };
}


export function generate<A>(gen: Gen<A>): A {
  let rng = new Prando(Math.random() * 1000);
  return gen(rng)(100);
}

export function property<A>(
  predicate: Predicate<A>,
  generator: Gen<A>,
  shrinker: Shrinker<A>): Property<A> {
  return (rng: Prando,
          size: number) => {
            let gen = generator(rng);
            let i = 0;
            for (; i < MAX_SUCCESS; i++) {
              let x = gen(size);
              if (!predicate(x)) {
                let counterexample = findMinimalCounterExample(x, predicate, shrinker);
                return { result: 'Falsified', tests: i+1, counterexample };
              }
              size++;
            }
            return { result: 'OK', tests: i, counterexample: null };
          };
}