X-Git-Url: http://git.scottworley.com/inverse-tax/blobdiff_plain/1c6e40691d86539d80018aa8fc495272b900c4c4..899824d5cc5289a3b15b22898fce2e11d1f63660:/tax.test.js

diff --git a/tax.test.js b/tax.test.js
index 48f2e0d..5080001 100644
--- a/tax.test.js
+++ b/tax.test.js
@@ -6,15 +6,18 @@ function test(description, f) {
   f();
 }
 
-function rand(limit = 100) {
+function rand(limit = 99) {
   return Math.round(Math.random() * limit);
 }
 
-function make_random_tax_table(min_threshold) {
-  if (rand(2)) return [];
-  const start = min_threshold === undefined ? rand() : min_threshold;
-  const end = start + 1 + rand();
-  return [[start, end, rand()]].concat(make_random_tax_table(end));
+function make_random_tax_table() {
+  function make_random_table(min_threshold) {
+    if (rand(4) == 0) return [];
+    const start = min_threshold === undefined ? rand() : min_threshold;
+    const end = start + 1 + rand();
+    return [[start, rand()]].concat(make_random_table(end));
+  }
+  return tax_table_from_table(make_random_table());
 }
 
 test("parse tax table", () => {
@@ -41,7 +44,7 @@ test("merge tax tables", () => {
     const combined = merge_tax_tables(t1, t2);
     for (var j = 0; j < 10; j++) {
       const income = rand(250);
-      assert.strictEqual(tax(t1, income) + tax(t2, income), tax(combined, income));
+      assert.ok(near(tax(t1, income) + tax(t2, income), tax(combined, income)));
     }
   }
 });
@@ -50,5 +53,18 @@ test("invert", () => {
   assert.strictEqual(invert([                  ])(10), 10);
   assert.strictEqual(invert([[ 0, Infinity, .1]])( 9), 10);
   assert.strictEqual(invert([[10, Infinity, .1]])(19), 20);
+  assert.strictEqual(invert([[10, Infinity, .1]])( 9),  9);
   assert.strictEqual(invert([[0, 100, .1], [100, Infinity, .2]])(170), 200);
+  assert.strictEqual(invert([[0, 100, .1], [100, Infinity, .2]])(112), 127.5);
+  assert.strictEqual(invert([[6, 90, 0.75], [90, Infinity, 0.12]])(27), 90);
+
+  for (var i = 0; i < 1000; i++) {
+    const t = make_random_tax_table();
+    const inverted_t = invert(t);
+    for (var j = 0; j < 20; j++) {
+      const net = rand(250);
+      const gross = inverted_t(net);
+      assert.ok(near(net, gross - tax(t, gross)));
+    }
+  }
 });