add scattergeo tests

Author: archmoj

test/jasmine/performance_tests/scattergeo_test.js

@@ -0,0 +1,204 @@
+var createGraphDiv = require('../assets/create_graph_div');
+var delay = require('../assets/delay');
+var Plotly = require('../../../lib/core');
+var PlotlyScattergeo = require('../../../lib/scattergeo');
+var downloadCSV = require('./assets/post_process').downloadCSV;
+var nSamples = require('./assets/constants').nSamples;
+
+Plotly.register(PlotlyScattergeo);
+
+var gd = createGraphDiv();
+
+const samples = Array.from({ length: nSamples }, (_, i) => i);
+const nTimes = samples.length - 1;
+
+var tests = [{
+    n: 1000, mode: 'markers', nTraces: 1
+}, {
+    n: 2000, mode: 'markers', nTraces: 1
+}, {
+    n: 4000, mode: 'markers', nTraces: 1
+}, {
+    n: 8000, mode: 'markers', nTraces: 1
+}, {
+    n: 16000, mode: 'markers', nTraces: 1
+}, {
+    n: 32000, mode: 'markers', nTraces: 1
+}, {
+    n: 64000, mode: 'markers', nTraces: 1
+}, {
+    n: 1000, mode: 'lines', nTraces: 1
+}, {
+    n: 2000, mode: 'lines', nTraces: 1
+}, {
+    n: 4000, mode: 'lines', nTraces: 1
+}, {
+    n: 8000, mode: 'lines', nTraces: 1
+}, {
+    n: 16000, mode: 'lines', nTraces: 1
+}, {
+    n: 32000, mode: 'lines', nTraces: 1
+}, {
+    n: 64000, mode: 'lines', nTraces: 1
+}, {
+    n: 1000, mode: 'markers+lines', nTraces: 1
+}, {
+    n: 2000, mode: 'markers+lines', nTraces: 1
+}, {
+    n: 4000, mode: 'markers+lines', nTraces: 1
+}, {
+    n: 8000, mode: 'markers+lines', nTraces: 1
+}, {
+    n: 16000, mode: 'markers+lines', nTraces: 1
+}, {
+    n: 32000, mode: 'markers+lines', nTraces: 1
+}, {
+    n: 64000, mode: 'markers+lines', nTraces: 1
+}, {
+    n: 1000, mode: 'markers', nTraces: 10
+}, {
+    n: 2000, mode: 'markers', nTraces: 10
+}, {
+    n: 4000, mode: 'markers', nTraces: 10
+}, {
+    n: 8000, mode: 'markers', nTraces: 10
+}, {
+    n: 16000, mode: 'markers', nTraces: 10
+}, {
+    n: 32000, mode: 'markers', nTraces: 10
+}, {
+    n: 64000, mode: 'markers', nTraces: 10
+}, {
+    n: 1000, mode: 'lines', nTraces: 10
+}, {
+    n: 2000, mode: 'lines', nTraces: 10
+}, {
+    n: 4000, mode: 'lines', nTraces: 10
+}, {
+    n: 8000, mode: 'lines', nTraces: 10
+}, {
+    n: 16000, mode: 'lines', nTraces: 10
+}, {
+    n: 32000, mode: 'lines', nTraces: 10
+}, {
+    n: 64000, mode: 'lines', nTraces: 10
+}, {
+    n: 1000, mode: 'markers+lines', nTraces: 10
+}, {
+    n: 2000, mode: 'markers+lines', nTraces: 10
+}, {
+    n: 4000, mode: 'markers+lines', nTraces: 10
+}, {
+    n: 8000, mode: 'markers+lines', nTraces: 10
+}, {
+    n: 16000, mode: 'markers+lines', nTraces: 10
+}, {
+    n: 32000, mode: 'markers+lines', nTraces: 10
+}, {
+    n: 64000, mode: 'markers+lines', nTraces: 10
+}, {
+    n: 1000, mode: 'markers', nTraces: 100
+}, {
+    n: 2000, mode: 'markers', nTraces: 100
+}, {
+    n: 4000, mode: 'markers', nTraces: 100
+}, {
+    n: 8000, mode: 'markers', nTraces: 100
+}, {
+    n: 16000, mode: 'markers', nTraces: 100
+}, {
+    n: 32000, mode: 'markers', nTraces: 100
+}, {
+    n: 64000, mode: 'markers', nTraces: 100
+}, {
+    n: 1000, mode: 'lines', nTraces: 100
+}, {
+    n: 2000, mode: 'lines', nTraces: 100
+}, {
+    n: 4000, mode: 'lines', nTraces: 100
+}, {
+    n: 8000, mode: 'lines', nTraces: 100
+}, {
+    n: 16000, mode: 'lines', nTraces: 100
+}, {
+    n: 32000, mode: 'lines', nTraces: 100
+}, {
+    n: 64000, mode: 'lines', nTraces: 100
+}, {
+    n: 1000, mode: 'markers+lines', nTraces: 100
+}, {
+    n: 2000, mode: 'markers+lines', nTraces: 100
+}, {
+    n: 4000, mode: 'markers+lines', nTraces: 100
+}, {
+    n: 8000, mode: 'markers+lines', nTraces: 100
+}, {
+    n: 16000, mode: 'markers+lines', nTraces: 100
+}, {
+    n: 32000, mode: 'markers+lines', nTraces: 100
+}, {
+    n: 64000, mode: 'markers+lines', nTraces: 100
+}];
+
+tests.forEach(function(spec, index) {
+    describe('Performance test ' + spec.nTraces + 'scattergeo | size:' + spec.n + ' | mode: ' + spec.mode, function() {
+        'use strict';
+
+        var startTime, endTime;
+
+        beforeEach(function(done) {
+            var y = Array.from({ length: spec.n }, (_, i) => 0.001 * i * Math.cos(Math.sqrt(i)));
+
+            var data = [];
+            var nPerTrace = Math.floor(spec.n / spec.nTraces);
+            for(var k = 0; k < spec.nTraces; k++) {
+                data.push({
+                    type: 'scattergeo',
+                    mode: spec.mode,
+                    lat: y.slice(k * nPerTrace, (k + 1) * nPerTrace),
+                    lon: Array.from({ length: nPerTrace }, (_, i) => -180 + 0.005 * (i + k * nPerTrace))
+                });
+            }
+
+            var mock = {
+                data: data,
+                layout: {
+                    showlegend: false,
+                    width: 900,
+                    height: 400
+                }
+            };
+
+            startTime = performance.now();
+
+            // Wait for actual rendering to complete
+            requestAnimationFrame(function() {
+                requestAnimationFrame(function() {
+                    endTime = performance.now();
+                    done();
+                });
+            });
+
+            Plotly.newPlot(gd, mock);
+        });
+
+        afterEach(function(done) {
+            delay(100)().then(done);
+        });
+
+        samples.forEach(function(t) {
+            it('should graph scattergeo traces | turn: ' + t, function() {
+                var delta = endTime - startTime;
+
+                if(t === 0) {
+                    tests[index].raw = [];
+                }
+                tests[index].raw[t] = delta;
+
+                if(t === nTimes && index === tests.length - 1) {
+                    downloadCSV('scattergeo', tests);
+                }
+            });
+        });
+    });
+});