init

.venv/lib/python3.8/site-packages/mypyc/test-data/fixtures/testutil.py

@@ -0,0 +1,52 @@
+# 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