Current File : //lib/python3/dist-packages/mypyc/test-data/fixtures/testutil.py
# Simple support library for our run tests.
from contextlib import contextmanager
from typing import (
Any, Iterator, TypeVar, Generator, Optional, List, Tuple, Sequence,
Union, Callable,
)
@contextmanager
def assertRaises(typ: type, msg: str = '') -> Iterator[None]:
try:
yield
except Exception as e:
assert isinstance(e, typ), "{} is not a {}".format(e, typ.__name__)
assert msg in str(e), 'Message "{}" does not match "{}"'.format(e, msg)
else:
assert False, "Expected {} but got no exception".format(typ.__name__)
T = TypeVar('T')
U = TypeVar('U')
V = TypeVar('V')
def run_generator(gen: Generator[T, V, U],
inputs: Optional[List[V]] = None,
p: bool = False) -> Tuple[Sequence[T], Union[U, str]]:
res: List[T] = []
i = -1
while True:
try:
if i >= 0 and inputs:
# ... fixtures don't have send
val = gen.send(inputs[i]) # type: ignore
else:
val = next(gen)
except StopIteration as e:
return (tuple(res), e.value)
except Exception as e:
return (tuple(res), str(e))
if p:
print(val)
res.append(val)
i += 1
F = TypeVar('F', bound=Callable)
# Wrap a mypyc-generated function in a real python function, to allow it to be
# stuck into classes and the like.
def make_python_function(f: F) -> F:
def g(*args: Any, **kwargs: Any) -> Any:
return f(*args, **kwargs)
return g # type: ignore
Mini Shell