updating generateSequenceData to pass annotation numbers

Author: tnrich

src/generateSequenceData.js

@@ -6,7 +6,11 @@ const objectid = require("bson-objectid");
 
 module.exports = function generateSequenceData({
   isProtein,
-  sequenceLength = 1000
+  sequenceLength = 1000,
+  numFeatures,
+  numParts,
+  numPrimers,
+  numTranslations
 } = {}) {
   let proteinSequence = isProtein && generateSequence(sequenceLength, true);
   let sequence = !isProtein && generateSequence(sequenceLength);
@@ -20,17 +24,27 @@ module.exports = function generateSequenceData({
     proteinSequence,
     translations: isProtein
       ? undefined
-      : generateAnnotations(5, 0, sequenceLength - 1, sequenceLength / 3),
+      : generateAnnotations(
+          numTranslations || 5,
+          0,
+          sequenceLength - 1,
+          sequenceLength / 3
+        ),
     features: generateAnnotations(
-      10,
+      numFeatures || 10,
       0,
       sequenceLength - 1,
       sequenceLength / 3
     ),
     primers: isProtein
       ? undefined
-      : generateAnnotations(10, 0, sequenceLength - 1, 50),
-    parts: generateAnnotations(10, 0, sequenceLength - 1, sequenceLength / 3)
+      : generateAnnotations(numPrimers || 10, 0, sequenceLength - 1, 50),
+    parts: generateAnnotations(
+      numParts || 10,
+      0,
+      sequenceLength - 1,
+      sequenceLength / 3
+    )
   };
 };
 

src/generateSequenceData.test.js

@@ -1,13 +1,22 @@
-const generateSequenceData = require('./generateSequenceData')
+const generateSequenceData = require("./generateSequenceData");
 const chai = require("chai");
-const { getRangeLength } = require("ve-range-utils");
 
 chai.should();
 const chaiSubset = require("chai-subset");
+const { map } = require("lodash");
 chai.use(chaiSubset);
 
-describe('generateSequenceData', function() {
-    it('should generate some nice random data', function() {
-      generateSequenceData({sequenceLength: 100}).sequence.length.should.equal(100)
-    })
-})
+describe("generateSequenceData", function() {
+  it("should generate some nice random data", function() {
+    generateSequenceData({ sequenceLength: 100 }).sequence.length.should.equal(
+      100
+    );
+  });
+  it("numFeatures should work", function() {
+    const a = generateSequenceData({
+      sequenceLength: 100,
+      numFeatures: 100
+    });
+    map(a.features).length.should.equal(100);
+  });
+});