@wizhut_tech/wizjs 0.1.3 → 0.1.4

package/__tests__/lang_itertools.test.js

@@ -1,6 +1,6 @@
 const t = require('tap');
 
-const { count, fn_count } = require('../src/lang/itertools.js');
+const { count, repeat } = require('../src/lang/itertools.js');
 
 
 t.test('itertools/count', (t) => {
@@ -19,3 +19,18 @@ t.test('itertools/count', (t) => {
 
     t.end();
 });
+
+
+t.test('itertools/repeat', (t) => {
+    // single repeat
+    const repeatGen = repeat(10, 1);
+    t.equal(repeatGen.next().value, 10);
+    t.equal(repeatGen.return().done, true);
+
+    // eternal repeat
+    const repeatGenTwo = repeat(15, 0);
+    t.equal(repeatGenTwo.next().value, 15);
+    t.equal(repeatGenTwo.next().value, 15);
+
+    t.end();
+});

package/docs/lang_itertools.md

@@ -1,4 +1,4 @@
 # Lang / itertools
 
 * **count(start, step=1)**: A generator that counts from *start* with a specified *step*. Default stepping is *1*.
-* **fn_count(start, step=1, fn)**: A generator that counts from *start* with a specified *step*. Default stepping is *1*. For each iteration the *fn* is called. To continue to the next iteration, the *fn* should return *true*. If not the iteration breaks,. 
+* **repeat(arg, times=0)**: A generator that repeats *arg* for *times* times. If times is 0 then it returns *arg* forever.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wizhut_tech/wizjs",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "main": "index.js",
   "devDependencies": {
     "tap": "21.1.0"

package/src/lang/itertools.js

@@ -1,10 +1,8 @@
-
-const {toInteger} = require("../math/numbers");
+const { toInteger } = require('../math/numbers.js');
 
 
 function* count(start=0, step=1) {
     let iCount = toInteger(start);
-    const iStep = toInteger(step);
 
     while (true) {
         yield iCount;
@@ -12,14 +10,28 @@ function* count(start=0, step=1) {
     }
 }
 
-function fn_count(start, fn, step=1) {
-    const countGenerator = count(start, step);
+function *repeat(arg, times=0) {
+    let iTimes = toInteger(times);
 
-    while (fn(countGenerator.next().value)) { }
-}
+    if (iTimes < 0) {
+        iTimes = 0;
+    }
+
+    let iTimesLeft = iTimes;
 
+    while (true) {
+        if (iTimes === 0) {
+            yield arg;
+        }
+
+        if (iTimesLeft > 0) {
+            iTimesLeft -= 1;
+            yield arg;
+        }
+    }
+}
 
 module.exports = {
     count,
-    fn_count
+    repeat
 };