Add tests for bare functions

Author: pganssle

tests/__init__.py

@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+
+"""Unit test package for variants."""

tests/_division_data.py

@@ -0,0 +1,46 @@
+
+class DivisionDataClass(object):
+    def __init__(self):
+        self.MODE_VALS = self.__get_mode_vals()
+
+    ALL_VALS = [(0, 1, 0)]
+
+    DIV_VALS = ALL_VALS + [
+        (12, 8, 1.50),
+        (11, 4, 2.75),
+        (21, 4, 5.25)
+    ]
+
+    ROUND_VALS = ALL_VALS + [
+        (12, 8, 2.00),
+        (11, 4, 3.00),
+        (21, 4, 5.00)
+    ]
+
+    FLOOR_VALS = ALL_VALS + [
+        (12, 8, 1.0),
+        (11, 4, 2.0),
+        (21, 4, 5.0)
+    ]
+
+    CEIL_VALS = ALL_VALS + [
+        (12, 8, 2.0),
+        (11, 4, 3.0),
+        (21, 4, 6.0)
+    ]
+
+    def __get_mode_vals(self):
+        modes_vals = [
+            (None, self.DIV_VALS),
+            ('round', self.ROUND_VALS),
+            ('floor', self.FLOOR_VALS),
+            ('ceil', self.CEIL_VALS)
+        ]
+
+        out = [(vals + (mode,))
+               for mode, all_vals in modes_vals
+               for vals in all_vals]
+
+        return out
+
+DivisionData = DivisionDataClass()

tests/test_functions.py

@@ -0,0 +1,92 @@
+from __future__ import division
+
+import math
+
+from variants import variants
+
+from ._division_data import DivisionData
+
+import pytest
+
+###
+# Example implementation - division function
+@variants
+def divide(x, y):
+    """A function that divides x by y."""
+    return x / y
+
+
+@divide.variant('round')
+def divide(x, y):
+    return round(x / y)
+
+
+@divide.variant('round_callmain')
+def divide(x, y):
+    return round(divide(x, y))
+
+
+@divide.variant('floor')
+def divide(x, y):
+    return math.floor(divide(x, y))
+
+
+@divide.variant('ceil')
+def divide(x, y):
+    return math.ceil(divide(x, y))
+
+
+@divide.variant('mode')
+def divide(x, y, mode=None):
+    funcs = {
+        None: divide,
+        'round': divide.round,
+        'floor': divide.floor,
+        'ceil': divide.ceil
+    }
+
+    return funcs[mode](x, y)
+
+###
+# Division Function Tests
+@pytest.mark.parametrize('x, y, expected', DivisionData.DIV_VALS)
+def test_main(x, y, expected):
+    assert divide(x, y) == expected
+
+
+@pytest.mark.parametrize('x,y,expected', DivisionData.ROUND_VALS)
+def test_round(x, y, expected):
+    assert divide.round(x, y) == expected
+
+
+@pytest.mark.parametrize('x,y,expected', DivisionData.ROUND_VALS)
+def test_round_callmain(x, y, expected):
+    assert divide.round_callmain(x, y) == expected
+
+
+@pytest.mark.parametrize('x,y,expected', DivisionData.FLOOR_VALS)
+def test_round_callmain(x, y, expected):
+    assert divide.floor(x, y) == expected
+
+@pytest.mark.parametrize('x,y,expected', DivisionData.CEIL_VALS)
+def test_round_callmain(x, y, expected):
+    assert divide.ceil(x, y) == expected
+
+
+@pytest.mark.parametrize('x,y,expected,mode', DivisionData.MODE_VALS)
+def test_mode(x, y, expected, mode):
+    assert divide.mode(x, y, mode) == expected
+
+
+###
+# Division function metadata tests
+def test_name():
+    assert divide.__name__ == 'divide'
+
+
+def test_docstring():
+    assert divide.__doc__ == """A function that divides x by y."""
+
+
+def test_repr():
+    assert repr(divide) == '<VariantFunction divide>'