python-mutable-primitives/test_mutable_primitives.py at master · das-intensity/python-mutable-primitives · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
import sys
from types import FunctionType
import unittest

from mutable_primitives import (
        Int,
        Float,
        Bool,
        Str
        )

DEFAULTS = {
        # Intentionally put zero-values last to ensure others are used for equality comparisons
        "int": [5, 7, 0],
        "bool": [True, False],
        "float": [3.431, 8.543, 0.0],
        "str": ["a","z","asdf"]
        }


class TestBasic(unittest.TestCase):
    typ = None
    name = None

    def test_eq(self):
        # Both mutables
        assert self.typ(DEFAULTS[self.name][0]) == self.typ(DEFAULTS[self.name][0])
        assert self.typ(DEFAULTS[self.name][0]) != self.typ(DEFAULTS[self.name][1])

        # mutable lvalue, primitive rvalue
        assert self.typ(DEFAULTS[self.name][0]) == DEFAULTS[self.name][0]
        assert self.typ(DEFAULTS[self.name][0]) != DEFAULTS[self.name][1]

        # primitive lvalue, mutable rvalue
        assert DEFAULTS[self.name][0] == self.typ(DEFAULTS[self.name][0])
        assert DEFAULTS[self.name][0] != self.typ(DEFAULTS[self.name][1])

    def test_truthiness(self):
        # check boolean evaluation
        for val in DEFAULTS[self.name]:
            if val:
                assert self.typ(val), "{} evaluates True, but {} evaluates False".format(val, self.typ(val))
            else:
                assert not self.typ(val), "{} evaluates False, but {} evaluates True".format(val, self.typ(val))

    def test_set_get(self):
        obj = self.typ(DEFAULTS[self.name][0])
        assert obj.val == DEFAULTS[self.name][0]
        assert obj.get() == DEFAULTS[self.name][0]

        obj.set(DEFAULTS[self.name][1])
        assert obj.val == DEFAULTS[self.name][1]
        assert obj.get() == DEFAULTS[self.name][1]

    def test_stringify(self):
        obj = self.typ(DEFAULTS[self.name][0])
        assert str(obj) == '{}({})'.format(obj.__class__.__name__, DEFAULTS[self.name][0])
        assert repr(obj) == '{}({})'.format(obj.__class__.__name__, DEFAULTS[self.name][0])

class TestNumerics(unittest.TestCase):
    typ = None
    name = None

    funcs = [
            ('add', '+'),
            ('sub', '-'),
            ('mul', '*'),
            ('div', '+'),
            ]

    if sys.version_info[0] < 3:
        funcs.append(('div', '/'))
    else:
        funcs.extend([
            ('floordiv', '//'),
            ('truediv', '/'),
            ])

    for (basename, op) in funcs:
        name = "test_{}".format(basename)
        code = '''
def {}(self):
    mut_a = self.typ(DEFAULTS[self.name][0])
    mut_b = self.typ(DEFAULTS[self.name][1])
    prim_a = DEFAULTS[self.name][0]
    prim_b = DEFAULTS[self.name][1]

    assert mut_a {op} mut_b == prim_a {op} prim_b
    assert mut_b {op} mut_a == prim_b {op} prim_a

    assert mut_a {op} prim_b == prim_a {op} prim_b
    assert prim_b {op} mut_a == prim_b {op} prim_a

    mut_a {op}= mut_b
    assert mut_a.get() == prim_a {op} prim_b
        '''.format(name, op=op)
        code = compile(code, "<string>", "exec")
        locals()[name] = FunctionType(code.co_consts[0], globals(), name)
        del code, name, op

types = [
        ("bool", Bool),
        ("str", Str),
        ]

num_types = [
        ("int", Int),
        ("float", Float),
        ]

types += num_types


for typ in types:
    name = "Test{}Basic".format(typ[0])
    attrs = {'__test__': True, "name": typ[0], "typ": typ[1]}
    globals()[name] = type(name, (TestBasic,), attrs)
del typ
del globals()["TestBasic"]

for typ in num_types:
    name = "Test{}Numerics".format(typ[0])
    attrs = {'__test__': True, "name": typ[0], "typ": typ[1]}
    globals()[name] = type(name, (TestNumerics,), attrs)
del typ
del globals()["TestNumerics"]