def test_Map_apply_multiple_args():
f = lambda x: lambda y: x + y
m = Map({'a': f})
n = Map({'a': 1})
o = Map({'a': 2})
p = m * n * o
assert p.v == {'a': 3}
def test_Map_apply_multiple_args_with_miss():
f = lambda x: lambda y: x + y
m = Map({'a': f})
n = Map({'a': 1})
o = Map({'b': 2})
p = Map({'a': 2})
q = m * n * o * p
assert q.v == {}
def test_Map_apply_only_matches():
f = lambda x: x + 1
g = lambda x: x - 1
h = lambda x: x * x
m = Map({'a': f, 'b': g, 'c': h})
n = Map({'a': 1, 'b': 2, 'c': 3})
o = m * n
assert o.v == {'a': 2, 'b': 1, 'c': 9}
def test_Map_is_monoidal():
m = Map({'a': 10})
n = Map({'b': 7})
o = Map({'c': 4})
assert Map.mappend(m, Map.mappend(n, o)) == \
Map.mappend(Map.mappend(m, n), o)
def test_Map_len():
assert len(Map({'a': 1})) == 1
def test_Map_mappend():
m = Map({'a': 10})
n = Map({'b': 7})
assert m.mappend(n) == Map({'a': 10, 'b': 7})
def test_Map_keys():
m = Map({'a': 1, 'b': 2, 'c': 3})
lm = ['a', 'b', 'c']
assert all(k in lm for k in m.keys())
@pytest.mark.parametrize('monoid, generic', [('', str), ([], Sequence),
({}, Mapping)])
def test_make_generic_mconcat_on_selected(monoid, generic):
assert m._make_generic_mconcat(monoid) is m._generic_mconcats[generic]
@pytest.mark.parametrize(
'objs, expected',
[(ints, 6), (list_of(float, ints), 6.0), (list_of(complex, ints), 6 + 0j),
(list_of(Decimal, ints), Decimal(6)),
(list_of(lambda a: {a}, ints), set(ints)),
(list_of(lambda a: frozenset([a]), ints), frozenset(ints)),
(list_of(lambda a: [a], ints), list(ints)), (list_of(str, ints), '123'),
(list_of(lambda a: {a: a}, ints), {
1: 1,
2: 2,
3: 3
}), (list_of(lambda a: (a, ), ints), list(ints)),
([True, False, True], True), ([False, False, False], False)])
def test_generic_mconcat(objs, expected):
assert m.generic_mconcat(*objs) == expected
@pytest.mark.parametrize('obj, is_monoidal',
[(Decimal(1), True), (List(), True), (1, True),
([], True), (object(), False), (Map(), True),
(DummyMonoid(), True), (True, True),
(WeakSet(), False)])
def test_is_monoid(obj, is_monoidal):
assert m.is_monoid(obj) == is_monoidal
def test_Map_mempty():
assert Map.mempty == Map()
def test_Map_fromkeys():
m = Map.fromkeys(['a', 'b', 'c'], 1)
assert isinstance(m, Map)
assert m.v == {'a': 1, 'b': 1, 'c': 1}
assert Map.fromkeys(['a']).v == {'a': None}
def test_Map_unit():
m = Map.unit(('a', 4))
assert m.v == {'a': 4}
def test_Map_mconcat():
m = Map({'a': 10})
n = Map({'b': 7})
o = Map({'c': 4})
assert Map.mconcat(m, n, o) == Map({'a': 10, 'b': 7, 'c': 4})
def test_Map_ne():
assert Map({'a': 1}) != Map({'a': 2})
assert Map({'a': 1}) != Map({'b': 1})
def test_Map_eq():
assert Map({'a': 1}) == Map({'a': 1})
def test_Map_contains():
assert 'a' in Map({'a': 1})
def test_Map_fmap():
f = lambda x: x + 1
m = Map({'a': 1, 'b': 2, 'c': 3})
assert m.fmap(f).v == {'a': 2, 'b': 3, 'c': 4}
assert m.fmap(identity).v == {'a': 1, 'b': 2, 'c': 3}
def test_Map_iter():
m = Map({'a': 1, 'b': 2, 'c': 3})
lm = ['a', 'b', 'c']
assert all(k in lm for k in iter(m))
def test_Map_apply():
m = Map.fromkeys('abc', lambda x: x + 1)
n = Map({'a': 1, 'b': 2, 'c': 3})
o = m * n
assert o.v == {'a': 2, 'b': 3, 'c': 4}
def test_Mempty_is_singleton():
assert Mempty is _Mempty()
def test_Mempty_mempty_is_Mempty():
assert Mempty.mempty is Mempty
def test_Mempty_is_false():
assert not Mempty
@pytest.mark.parametrize(
'monoid', [[4], [{1, 2}], [List(1, 2)], [Map({'a': 10})], [True],
pytest.mark.xfail([Decimal('0')])])
def test_Mempty_mappend(monoid):
assert Mempty.mappend(monoid) == monoid
@pytest.mark.parametrize('monoids, result',
[((1, 2, 3), 6), ((1, Mempty, 2, 3), 6),
([[1], [2], [3]], [1, 2, 3]),
([Mempty, [1], [2], [3]], [1, 2, 3])])
def test_Mempty_mconcat(monoids, result):
assert Mempty.mconcat(*monoids) == result
def test_Mempty_mconcat_with_all_mempties():
assert Mempty.mconcat(Mempty, Mempty) is Mempty
def test_Map_getitem():
m = Map({'a': 1})
assert m['a'] == 1