From 4af498408f1dcd8fdbb746667243160c2041fa56 Mon Sep 17 00:00:00 2001
From: archmoj <mojtaba@plot.ly>
Date: Fri, 18 Oct 2019 16:17:19 -0400
Subject: [PATCH] simplify get direction function in streamtube calc and
 improve jasmine test for random input

---
 src/traces/streamtube/calc.js         |  2 +-
 test/jasmine/tests/streamtube_test.js | 22 +++++++++++-----------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/src/traces/streamtube/calc.js b/src/traces/streamtube/calc.js
index 2ea9e8b0f20..5b695c3e343 100644
--- a/src/traces/streamtube/calc.js
+++ b/src/traces/streamtube/calc.js
@@ -118,7 +118,7 @@ module.exports = function calc(gd, trace) {
 
     var getArray = function(c) { return c === 'x' ? x : c === 'y' ? y : z; };
     var getVals = function(c) { return c === 'x' ? Xs : c === 'y' ? Ys : Zs; };
-    var getDir = function(c) { return (+(c[len - 1] - c[0])) * 2 + 1; };
+    var getDir = function(c) { return (c[len - 1] < c[0]) ? -1 : 1; };
 
     var arrK = getArray(gridFill[1]);
     var arrJ = getArray(gridFill[3]);
diff --git a/test/jasmine/tests/streamtube_test.js b/test/jasmine/tests/streamtube_test.js
index 94135044e6a..ac8d1f8e7ec 100644
--- a/test/jasmine/tests/streamtube_test.js
+++ b/test/jasmine/tests/streamtube_test.js
@@ -305,17 +305,13 @@ describe('Test streamtube interactions', function() {
         var v = [];
         var w = [];
 
-        for(var i = 0; i < 3; i++) {
-            for(var j = i; j < 4; j++) {
-                for(var k = j; k < 5; k++) {
-                    x.push(i);
-                    y.push(j);
-                    z.push(k);
-                    u.push(1);
-                    v.push(1);
-                    w.push(1);
-                }
-            }
+        for(var n = 0; n < 1000; n++) {
+            x.push((10 * Math.random()) | 0);
+            y.push((10 * Math.random()) | 0);
+            z.push((10 * Math.random()) | 0);
+            u.push(1);
+            v.push(1);
+            w.push(1);
         }
 
         var fig = {
@@ -338,11 +334,15 @@ describe('Test streamtube interactions', function() {
             expect(exp.cellsLength).toBe(objs[0].cells.length, 'cells length - ' + msg);
         }
 
+        spyOn(Lib, 'warn');
+
         Plotly.plot(gd, fig).then(function() {
             _assert('arbitrary coordinates', {
                 positionsLength: 0,
                 cellsLength: 0
             });
+        }).then(function() {
+            expect(Lib.warn).toHaveBeenCalledWith('Encountered arbitrary coordinates! Unable to input data grid.');
         })
         .catch(failTest)
         .then(done);