1 files changed, 64 insertions, 6 deletions

diff --git a/pbt/ts/src/index.ts b/pbt/ts/src/index.ts
index 46454f0..474b656 100644
--- a/pbt/ts/src/index.ts
+++ b/pbt/ts/src/index.ts
@@ -1,12 +1,19 @@
 import "dotenv/config";
 import Prando from 'prando';
-import { Gen } from "./property";
+import { Gen, property } from "./property";
 
 let genint: Gen<number> = (rng: Prando) =>
   (size: number) =>
-  rng.nextInt(-size, size);
+    rng.nextInt(-size, size);
 
 
+function arrayEquals<A>(a: A[], b: A[]): boolean {
+  return Array.isArray(a) &&
+    Array.isArray(b) &&
+    a.length === b.length &&
+    a.every((val, index) => val === b[index]);
+}
+
 function genlist<A>(gen: Gen<A>): Gen<A[]> {
   return (rng: Prando) => {
     let g = gen(rng);
@@ -20,13 +27,64 @@ function genlist<A>(gen: Gen<A>): Gen<A[]> {
   };
 }
 
-function generate<A>(gen: Gen<A>): A {
-  let rng = new Prando(Math.random() * 1000);
-  return gen(rng)(100);
+function reverse<A>(arr: A[]): A[] {
+  let result = [];
+  for (let i = arr.length - 1; i >= 0; i--) {
+    result.push(arr[i]);
+  }
+  return result;
+}
+
+function reverse_is_self_inverse(arr: number[]): boolean {
+  return arrayEquals(reverse(reverse(arr)), arr);
+}
+
+function shrinklist<A>(arr: A[]): A[][] {
+  let result = [];
+  for (let i = 0; i < arr.length; i++) {
+    let copy = arr.slice();
+    copy.splice(i, 1);
+    result.push(copy);
+  }
+  return result;
+}
+
+function gen2lists<A>(gen: Gen<A>): Gen<[A[], A[]]> {
+  return (rng: Prando) => {
+    let gl = genlist(gen)(rng);
+    return (size: number) => {
+      return [gl(size), gl(size)]
+    };
+  };
+}
+
+function reverse_with_two_lists<A>(arrs: [A[], A[]]): boolean {
+  return arrayEquals(reverse(arrs[0].concat(arrs[1])),
+    reverse(arrs[0]).concat(reverse(arrs[1])));
 }
 
+function shrink2lists<A>([l1, l2]: [A[], A[]]): [A[], A[]][] {
+  let result: [A[], A[]][] = [];
+  let s1 = shrinklist(l1);
+  let s2 = shrinklist(l2);
+  for (let i = 0; i < s1.length; i++) {
+    let j = s2.length - i -1;
+    let news2 = j >= 0 ? s2[j] : [];
+    result.push([s1[i], news2]);
+  }
+  return result;
+}
+
+let prop_reverse_with_two_lists =
+  property<[number[], number[]]>(reverse_with_two_lists, gen2lists(genint), shrink2lists);
+
+let prop_reverse_is_self_inverse =
+  property(reverse_is_self_inverse, genlist(genint), shrinklist);
+
 async function main() {
-  console.log('list: ' + generate(genlist(genint)));
+  let rng = new Prando(Math.random() * 1000);
+  let result = prop_reverse_with_two_lists(rng, 10);
+  console.log('result: ' + JSON.stringify(result));
 }
 
 main();