Added load test showing a large number of plots

Author: PierreRaybaut

qwt/tests/curvebenchmark1.py

@@ -46,11 +46,16 @@ def get_curve_color():
     return colors[COLOR_INDEX]
 
 
+PLOT_ID = 0
+
+
 class BMPlot(QwtPlot):
     def __init__(self, title, xdata, ydata, style, symbol=None, *args):
         super(BMPlot, self).__init__(*args)
-        self.setMinimumSize(200, 200)
-        self.setTitle(title)
+        global PLOT_ID
+        self.setMinimumSize(200, 150)
+        PLOT_ID += 1
+        self.setTitle("%s (#%d)" % (title, PLOT_ID))
         self.setAxisTitle(QwtPlot.xBottom, "x")
         self.setAxisTitle(QwtPlot.yLeft, "y")
         self.curve_nb = 0
@@ -69,11 +74,11 @@ def __init__(self, title, xdata, ydata, style, symbol=None, *args):
 
 
 class BMWidget(QWidget):
-    def __init__(self, points, *args, **kwargs):
+    def __init__(self, nbcol, points, *args, **kwargs):
         super(BMWidget, self).__init__()
         self.plot_nb = 0
         self.curve_nb = 0
-        self.setup(points, *args, **kwargs)
+        self.setup(nbcol, points, *args, **kwargs)
 
     def params(self, *args, **kwargs):
         if kwargs.get("only_lines", False):
@@ -84,11 +89,11 @@ def params(self, *args, **kwargs):
                 ("Dots", None),
             )
 
-    def setup(self, points, *args, **kwargs):
+    def setup(self, nbcol, points, *args, **kwargs):
         x = np.linspace(0.001, 20.0, int(points))
         y = (np.sin(x) / x) * np.cos(20 * x)
         layout = QGridLayout()
-        nbcol, col, row = 2, 0, 0
+        col, row = 0, 0
         for style, symbol in self.params(*args, **kwargs):
             plot = BMPlot(style, x, y, getattr(QwtPlotCurve, style), symbol=symbol)
             layout.addWidget(plot, row, col)
@@ -102,7 +107,7 @@ def setup(self, points, *args, **kwargs):
         self.text.setReadOnly(True)
         self.text.setAlignment(Qt.AlignCenter)
         self.text.setText("Rendering plot...")
-        layout.addWidget(self.text, row + 1, 0, 1, 2)
+        layout.addWidget(self.text, row + 1, 0, 1, nbcol)
         self.setLayout(layout)
 
 
@@ -171,7 +176,7 @@ def run_benchmark(self, max_n, **kwargs):
         for idx in range(4, -1, -1):
             points = int(max_n / 10 ** idx)
             t0 = time.time()
-            widget = BMWidget(points, **kwargs)
+            widget = BMWidget(2, points, **kwargs)
             title = "%d points" % points
             description = "%d plots with %d curves of %d points" % (
                 widget.plot_nb,

qwt/tests/curvebenchmark2.py

@@ -74,7 +74,7 @@ def run_benchmark(self, max_n, **kwargs):
         ):
             t0 = time.time()
             symtext = "with%s symbols" % ("" if symbols else "out")
-            widget = CSWidget(points, symbols, **kwargs)
+            widget = CSWidget(2, points, symbols, **kwargs)
             title = "%d points" % points
             description = "%d plots with %d curves of %d points, %s" % (
                 widget.plot_nb,

qwt/tests/data/loadtest.png

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+#
+# Licensed under the terms of the MIT License
+# Copyright (c) 2015-2021 Pierre Raybaut
+# (see LICENSE file for more details)
+
+"""Load test"""
+
+SHOW = True  # Show test in GUI-based test launcher
+
+import time
+import os
+
+# Local imports
+from qwt.tests import curvebenchmark1 as cb
+
+
+if os.environ.get("USE_PYQWT5", False):
+    USE_PYQWT5 = True
+    from PyQt4.Qwt5 import QwtPlot, QwtPlotCurve
+else:
+    USE_PYQWT5 = False
+    from qwt import QwtPlot, QwtPlotCurve  # analysis:ignore
+
+
+NCOLS, NROWS = 6, 5
+NPLOTS = NCOLS * NROWS * 5 * 3
+
+
+class LTWidget(cb.BMWidget):
+    def params(self, *args, **kwargs):
+        return tuple([("Lines", None)] * NCOLS * NROWS)
+
+
+class LoadTest(cb.CurveBenchmark1):
+    TITLE = "Load test [%d plots]" % NPLOTS
+    SIZE = (1600, 700)
+
+    def __init__(self, max_n=100, parent=None, unattended=False, **kwargs):
+        super(LoadTest, self).__init__(
+            max_n=max_n, parent=parent, unattended=unattended, **kwargs
+        )
+
+    def run_benchmark(self, max_n, **kwargs):
+        points, symbols = 100, False
+        for _i_page in range(int(NPLOTS / (NCOLS * NROWS))):
+            t0 = time.time()
+            symtext = "with%s symbols" % ("" if symbols else "out")
+            widget = LTWidget(NCOLS, points, symbols, **kwargs)
+            title = "%d points" % points
+            description = "%d plots with %d curves of %d points, %s" % (
+                widget.plot_nb,
+                widget.curve_nb,
+                points,
+                symtext,
+            )
+            self.process_iteration(title, description, widget, t0)
+
+
+if __name__ == "__main__":
+    from qwt.tests import test_widget
+
+    app = test_widget(LoadTest, options=False)