Implement keepHigh

1 files changed, 29 insertions, 1 deletions

diff --git a/__tests__/dice.test.js b/__tests__/dice.test.js
index 1274b3d..1be0744 100644
--- a/__tests__/dice.test.js
+++ b/__tests__/dice.test.js
@@ -5,7 +5,8 @@ const {
   add,
   subtract,
   negative,
-  explode
+  explode,
+  keepHigh
 } = require('../src/dice.js')
 
 const defaultNumberRolls = 500
@@ -378,3 +379,30 @@ describe('exploding dice', () => {
     })
   })
 })
+
+describe('keep', () => {
+  describe('1K2d20', () => {
+    const die = keepHigh(constant(1), d(constant(2), constant(20)))
+
+    testDie(die, {
+      diceCount: 1,
+      average: {
+        average: 13.825
+      },
+      variance: {
+        variance: 22.104
+      },
+      bounds: {
+        low: 1,
+        high: 20,
+        expectLow: true,
+        expectHigh: true
+      }
+    }, 800)
+  })
+
+  describe('2K1d20', () => {
+    const die = keepHigh(constant(2), d(constant(1), constant(20)))
+    testDie(die, basicDieTestSpecs(1, 20))
+  })
+})