tests/oweditdomain: Add tests for utilities

Author: ales-erjavec

Orange/widgets/data/tests/test_oweditdomain.py

@@ -1,12 +1,13 @@
 # Test methods with long descriptive names can omit docstrings
 # pylint: disable=all
-from orangewidget.tests.utils import simulate
-from orangewidget.utils.itemmodels import PyListModel
-
+import pickle
 from unittest import TestCase
 import numpy as np
 from numpy.testing import assert_array_equal
 
+from orangewidget.tests.utils import simulate
+from orangewidget.utils.itemmodels import PyListModel
+
 from AnyQt.QtCore import QModelIndex, QItemSelectionModel, Qt, QItemSelection
 from AnyQt.QtWidgets import QAction, QComboBox, QLineEdit, QStyleOptionViewItem
 from AnyQt.QtTest import QTest, QSignalSpy
@@ -25,14 +26,16 @@
     AsString, AsCategorical, AsContinuous, AsTime,
     table_column_data, ReinterpretVariableEditor, CategoricalVector,
     VariableEditDelegate, TransformRole,
-    RealVector, TimeVector, StringVector)
+    RealVector, TimeVector, StringVector, make_dict_mapper, DictMissingConst,
+    LookupMappingTransform, as_float_or_nan)
 from Orange.widgets.data.owcolor import OWColor, ColorRole
 from Orange.widgets.tests.base import WidgetTest, GuiTest
 from Orange.tests import test_filename, assert_array_nanequal
 
 
 MArray = np.ma.MaskedArray
 
+
 class TestReport(TestCase):
     def test_rename(self):
         var = Real("X", (-1, ""), ())
@@ -676,3 +679,67 @@ def test_reinterpret_string(self):
                 [1.0, 1., np.nan, "1.0", 2020., 1., 1577836800., "2020"],
             ], dtype=object)
         )
+
+
+class TestUtils(TestCase):
+    def test_mapper(self):
+        mapper = make_dict_mapper({"a": 1, "b": 2})
+        r = mapper(["a", "a", "b"])
+        assert_array_equal(r, [1, 1, 2])
+        self.assertEqual(r.dtype, np.dtype("O"))
+        r = mapper(["a", "a", "b"], dtype=float)
+        assert_array_equal(r, [1, 1, 2])
+        self.assertEqual(r.dtype, np.dtype(float))
+        r = mapper(["a", "a", "b"], dtype=int)
+        self.assertEqual(r.dtype, np.dtype(int))
+
+        mapper = make_dict_mapper({"a": 1, "b": 2}, dtype=int)
+        r = mapper(["a", "a", "b"])
+        self.assertEqual(r.dtype, np.dtype(int))
+
+        r = np.full(3, -1, dtype=float)
+        r_ = mapper(["a", "a", "b"], out=r)
+        self.assertIs(r, r_)
+        assert_array_equal(r, [1, 1, 2])
+
+    def test_dict_missing(self):
+        d = DictMissingConst("<->", {1: 1, 2: 2})
+        self.assertEqual(d[1], 1)
+        self.assertEqual(d[-1], "<->")
+        # must be sufficiently different from defaultdict to warrant existence
+        self.assertEqual(d, {1: 1, 2: 2})
+
+    def test_as_float_or_nan(self):
+        a = np.array(["a", "1.1", ".2", "NaN"], object)
+        r = as_float_or_nan(a)
+        assert_array_equal(r, [np.nan, 1.1, .2, np.nan])
+
+        a = np.array([1, 2, 3], dtype=int)
+        r = as_float_or_nan(a)
+        assert_array_equal(r, [1., 2., 3.])
+
+        r = as_float_or_nan(r, dtype=np.float32)
+        assert_array_equal(r, [1., 2., 3.])
+        self.assertEqual(r.dtype, np.dtype(np.float32))
+
+
+class TestLookupMappingTransform(TestCase):
+    def setUp(self) -> None:
+        self.lookup = LookupMappingTransform(
+            StringVariable("S"),
+            DictMissingConst(np.nan, {"": np.nan, "a": 0, "b": 1}),
+            dtype=float,
+        )
+
+    def test_transform(self):
+        r = self.lookup.transform(np.array(["", "a", "b", "c"]))
+        assert_array_equal(r, [np.nan, 0, 1, np.nan])
+
+    def test_pickle(self):
+        lookup = self.lookup
+        lookup_ = pickle.loads(pickle.dumps(lookup))
+        c = np.array(["", "a", "b", "c"])
+        r = lookup.transform(c)
+        assert_array_equal(r, [np.nan, 0, 1, np.nan])
+        r_ = lookup_.transform(c)
+        assert_array_equal(r_, [np.nan, 0, 1, np.nan])