Add unit tests

stephane-caron · stephane-caron · commit 3682c26fcad3 · 2024-10-29T17:33:20.000+01:00
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -0,0 +1 @@
+# This file makes `coverage run -m unittest discover` discover tests.
diff --git a/tests/test_color_picker.py b/tests/test_color_picker.py
@@ -0,0 +1,21 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2023 Inria
+
+import unittest
+
+from uplot.color_picker import ColorPicker
+
+
+class TestColorPicker(unittest.TestCase):
+    def test_color_picker(self):
+        picker = ColorPicker()
+        first_color = picker.get_next_color()
+        second_color = picker.get_next_color()
+        self.assertNotEqual(first_color, second_color)
+
+        picker.reset()
+        reset_color = picker.get_next_color()
+        self.assertEqual(first_color, reset_color)
diff --git a/tests/test_generate_html.py b/tests/test_generate_html.py
@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2023 Inria
+
+import unittest
+
+import numpy as np
+
+from uplot.generate_html import generate_html
+
+
+class TestGenerateHtml(unittest.TestCase):
+    def setUp(self):
+        t = np.linspace(0.0, 1.0, 10)
+        self.data = [t, np.exp(0.42 * t)]
+        self.t = t
+
+    def test_generate_html(self):
+        test_title = "Test plot"
+        opts = {
+            "width": 1921,
+            "height": 601,
+            "title": test_title,
+            "scales": {"x": {"time": False}},
+            "series": [{}, {"stroke": "red"}],
+        }
+        html = generate_html(
+            opts,
+            self.data,
+            resize=False,
+        )
+        self.assertIn("1921", html)
+        self.assertIn("601", html)
+        self.assertIn(test_title, html)
+
+    def test_resize(self):
+        opts = {"series": [{}, {"stroke": "red"}]}
+        html = generate_html(opts, self.data, resize=True)
+        self.assertIn('window.addEventListener("resize', html)
diff --git a/tests/test_plot.py b/tests/test_plot.py
@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2023 Inria
+
+import unittest
+
+import numpy as np
+
+import uplot
+
+
+class TestPlot(unittest.TestCase):
+    def test_plot(self):
+        t = np.linspace(0.0, 1.0, 10)
+        data = [
+            t,
+            np.exp(0.42 * t),
+        ]
+        opts = {
+            "width": 1920,
+            "height": 600,
+            "title": "Simple plot",
+            "scales": {
+                "x": {
+                    "time": False,
+                },
+            },
+            "series": [
+                {},
+                {
+                    "stroke": "red",
+                    "fill": "rgba(255,0,0,0.1)",
+                },
+            ],
+        }
+        uplot.plot(opts, data)
diff --git a/tox.ini b/tox.ini
@@ -0,0 +1,43 @@
+[tox]
+isolated_build = True
+envlist =
+    coverage
+    lint
+    py{38,39,310,311,312}-{linux,macos,windows}
+
+[gh-actions]
+python =
+    3.8: py38
+    3.9: py39
+    3.10: py310
+    3.11: py311
+    3.12: py312
+
+[testenv]
+deps =
+    pytest >=7.1.2
+commands =
+    pytest tests
+
+[testenv:coverage]
+deps =
+    coverage >=5.5
+commands =
+    coverage erase
+    coverage run -m unittest discover
+    coverage report --include="uplot/**"
+
+[testenv:lint]
+deps =
+    black >=22.10.0
+    mypy >=1.10.0
+    pylint >=2.8.2
+    pytype >=2023.5.24
+    ruff >=0.4.3
+    types-setuptools >=65.6.0.2
+commands =
+    black --check --diff uplot
+    mypy uplot --ignore-missing-imports
+    pylint uplot --exit-zero --rcfile={toxinidir}/tox.ini
+    pytype uplot
+    ruff check uplot

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	+# This file makes `coverage run -m unittest discover` discover tests.