mirror of
https://github.com/correl/typesafe-monads.git
synced 2024-11-14 11:09:36 +00:00
102 lines
3 KiB
Python
102 lines
3 KiB
Python
import pytest # type: ignore
|
|
from typing import Any, Callable, List, TypeVar
|
|
|
|
from monads import Functor, Applicative, Future
|
|
|
|
T = TypeVar("T")
|
|
S = TypeVar("S")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_types() -> None:
|
|
m: Future[int] = Future.pure(1)
|
|
await m
|
|
map: Future[int] = Future.pure(1).map(lambda x: x)
|
|
await map
|
|
map_operator: Future[int] = Future.pure(1) * (lambda x: x)
|
|
await map_operator
|
|
bind: Future[int] = Future.pure(1).bind(lambda x: Future.pure(x))
|
|
await bind
|
|
bind_operator: Future[int] = Future.pure(1) >> (lambda x: Future.pure(x))
|
|
await bind_operator
|
|
apply: Future[int] = Future.pure(1).apply(Future.pure(lambda x: x))
|
|
await apply
|
|
apply_operator: Future[int] = Future.pure(lambda x: x) & Future.pure(1)
|
|
await apply_operator
|
|
sequence: Future[List[int]] = Future.sequence([Future.pure(1)])
|
|
await sequence
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_functor_identity() -> None:
|
|
identity: Callable[[int], int] = lambda x: x
|
|
assert await Future.pure(3) == await Future.pure(3).map(identity)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_functor_associativity() -> None:
|
|
f: Callable[[int], int] = lambda x: x + 1
|
|
g: Callable[[int], str] = lambda x: str(x)
|
|
assert await Future.pure(3).map(lambda x: g(f(x))) == await Future.pure(3).map(
|
|
f
|
|
).map(g)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_functor_map_mul_operator() -> None:
|
|
identity: Callable[[int], int] = lambda x: x
|
|
assert await Future.pure(3).map(identity) == await (Future.pure(3) * identity)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_functor_map_rmul_operator() -> None:
|
|
identity: Callable[[int], int] = lambda x: x
|
|
assert await Future.pure(3).map(identity) == await (identity * Future.pure(3))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_applicative_fmap_using_ap() -> None:
|
|
f: Callable[[int], int] = lambda x: x + 1
|
|
assert await Future.pure(3).map(f) == await Future.pure(3).apply(Future.pure(f))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_monad_bind() -> None:
|
|
expected: Future[int] = Future.pure(2)
|
|
m: Future[int] = Future.pure(1)
|
|
assert await expected == await m.bind(lambda x: Future.pure(x + 1))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_monad_bind_rshift_operator() -> None:
|
|
f: Callable[[int], Future[int]] = lambda x: Future.pure(x + 1)
|
|
assert await Future.pure(2).bind(f) == await (Future.pure(2) >> f)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_monad_left_identity() -> None:
|
|
n: int = 3
|
|
|
|
def f(n: int) -> Future[int]:
|
|
return Future.pure(n * 3)
|
|
|
|
m: Future[int] = Future.pure(n)
|
|
assert await m.bind(f) == await f(n)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_monad_right_identity() -> None:
|
|
assert await Future.pure(3) == await Future.pure(3).bind(lambda x: Future.pure(x))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_monad_associativity() -> None:
|
|
def f(n: int) -> Future[int]:
|
|
return Future.pure(n * 3)
|
|
|
|
def g(n: int) -> Future[int]:
|
|
return Future.pure(n + 5)
|
|
|
|
assert await Future.pure(3).bind(f).bind(g) == await Future.pure(3).bind(
|
|
lambda x: f(x).bind(g)
|
|
)
|