1 |
|
2 | "use strict";
|
3 |
|
4 | var assert = require("assert");
|
5 | var either = require("./either.js");
|
6 | var lazyseq = require("lazy-seq");
|
7 | var sum = require("./sum.js");
|
8 | var utils = require("./utils.js");
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 | function shrinkProtoIsoMap(f, g) {
|
26 |
|
27 | var shrink = this;
|
28 | return shrinkBless(function (value) {
|
29 | return shrink(g(value)).map(f);
|
30 | });
|
31 | }
|
32 |
|
33 |
|
34 |
|
35 |
|
36 |
|
37 |
|
38 |
|
39 |
|
40 |
|
41 |
|
42 |
|
43 |
|
44 |
|
45 |
|
46 |
|
47 |
|
48 |
|
49 | function shrinkBless(shrink) {
|
50 | shrink.smap = shrinkProtoIsoMap;
|
51 | return shrink;
|
52 | }
|
53 |
|
54 |
|
55 |
|
56 |
|
57 | var shrinkNoop = shrinkBless(function shrinkNoop() {
|
58 | return [];
|
59 | });
|
60 |
|
61 |
|
62 |
|
63 |
|
64 | function shrinkPair(shrinkA, shrinkB) {
|
65 | var result = shrinkBless(function (pair) {
|
66 | assert(pair.length === 2, "shrinkPair: pair should be an Array of length 2");
|
67 |
|
68 | var a = pair[0];
|
69 | var b = pair[1];
|
70 |
|
71 | var shrinkedA = shrinkA(a);
|
72 | var shrinkedB = shrinkB(b);
|
73 |
|
74 | var pairA = shrinkedA.map(function (ap) {
|
75 | return [ap, b];
|
76 | });
|
77 |
|
78 | if (Array.isArray(pairA)) {
|
79 | pairA = lazyseq.fromArray(pairA);
|
80 | }
|
81 |
|
82 | return pairA.append(function () {
|
83 | var pairB = shrinkedB.map(function (bp) {
|
84 | return [a, bp];
|
85 | });
|
86 | return pairB;
|
87 | });
|
88 | });
|
89 |
|
90 | return utils.curried3(result, arguments);
|
91 | }
|
92 |
|
93 |
|
94 |
|
95 |
|
96 | function shrinkEither(shrinkA, shrinkB) {
|
97 | function shrinkLeft(value) {
|
98 | return shrinkA(value).map(either.left);
|
99 | }
|
100 |
|
101 | function shrinkRight(value) {
|
102 | return shrinkB(value).map(either.right);
|
103 | }
|
104 |
|
105 | var result = shrinkBless(function (e) {
|
106 | return e.either(shrinkLeft, shrinkRight);
|
107 | });
|
108 |
|
109 | return utils.curried3(result, arguments);
|
110 | }
|
111 |
|
112 |
|
113 |
|
114 |
|
115 | function fromLinkedList(ll) {
|
116 | assert(ll.length === 1 || ll.length === 2, "linked list must be either [] or [x, linkedlist]");
|
117 | if (ll.length === 1) {
|
118 | return [ll[0]];
|
119 | } else {
|
120 | return [ll[0]].concat(fromLinkedList(ll[1]));
|
121 | }
|
122 | }
|
123 |
|
124 | function toLinkedList(arr) {
|
125 | assert(Array.isArray(arr) && arr.length > 0, "toLinkedList expects non-empty array");
|
126 | if (arr.length === 1) {
|
127 | return [arr[0]];
|
128 | } else {
|
129 | return [arr[0], toLinkedList(arr.slice(1))];
|
130 | }
|
131 | }
|
132 |
|
133 | function toSingleton(x) {
|
134 | return [x];
|
135 | }
|
136 |
|
137 |
|
138 | function fromSingleton(a) {
|
139 | return a[0];
|
140 | }
|
141 |
|
142 | function flattenShrink(shrinksLL) {
|
143 | if (shrinksLL.length === 1) {
|
144 | return shrinksLL[0].smap(toSingleton, fromSingleton);
|
145 | } else {
|
146 | var head = shrinksLL[0];
|
147 | var tail = shrinksLL[1];
|
148 | return shrinkPair(head, flattenShrink(tail));
|
149 | }
|
150 | }
|
151 |
|
152 |
|
153 |
|
154 |
|
155 | function shrinkTuple(shrinks) {
|
156 | assert(shrinks.length > 0, "shrinkTuple needs > 0 values");
|
157 | var shrinksLL = toLinkedList(shrinks);
|
158 | var shrink = flattenShrink(shrinksLL);
|
159 | var result = shrinkBless(function (tuple) {
|
160 | assert(tuple.length === shrinks.length, "shrinkTuple: not-matching params");
|
161 | var ll = toLinkedList(tuple);
|
162 | return shrink(ll).map(fromLinkedList);
|
163 | });
|
164 |
|
165 | return utils.curried2(result, arguments);
|
166 | }
|
167 |
|
168 |
|
169 |
|
170 |
|
171 | function shrinkSum(shrinks) {
|
172 | assert(shrinks.length > 0, "shrinkTuple needs > 0 values");
|
173 | var result = shrinkBless(function (s) {
|
174 | return s.fold(function (idx, len, value) {
|
175 | assert(len === shrinks.length, "shrinkSum: not-matching params");
|
176 | return shrinks[idx](value).map(function (shrinked) {
|
177 | return sum.addend(idx, len, shrinked);
|
178 | });
|
179 | });
|
180 | });
|
181 |
|
182 | return utils.curried2(result, arguments);
|
183 | }
|
184 |
|
185 | function shrinkArrayWithMinimumSize(size) {
|
186 | function shrinkArrayImpl(shrink) {
|
187 | var result = shrinkBless(function (arr) {
|
188 | assert(Array.isArray(arr), "shrinkArrayImpl() expects array, got: " + arr);
|
189 | if (arr.length <= size) {
|
190 | return lazyseq.nil;
|
191 | } else {
|
192 | var x = arr[0];
|
193 | var xs = arr.slice(1);
|
194 |
|
195 | return lazyseq.cons(xs, lazyseq.nil)
|
196 | .append(shrink(x).map(function (xp) { return [xp].concat(xs); }))
|
197 | .append(shrinkArrayImpl(shrink, xs).map(function (xsp) { return [x].concat(xsp); }));
|
198 | }
|
199 | });
|
200 |
|
201 | return utils.curried2(result, arguments);
|
202 | }
|
203 |
|
204 | return shrinkArrayImpl;
|
205 | }
|
206 |
|
207 |
|
208 |
|
209 |
|
210 | var shrinkArray = shrinkArrayWithMinimumSize(0);
|
211 |
|
212 |
|
213 |
|
214 |
|
215 | var shrinkNEArray = shrinkArrayWithMinimumSize(1);
|
216 |
|
217 | module.exports = {
|
218 | noop: shrinkNoop,
|
219 | pair: shrinkPair,
|
220 | either: shrinkEither,
|
221 | tuple: shrinkTuple,
|
222 | sum: shrinkSum,
|
223 | array: shrinkArray,
|
224 | nearray: shrinkNEArray,
|
225 | bless: shrinkBless,
|
226 | };
|