okstra 0.113.0 → 0.115.0

package/runtime/python/okstra_vendor/networkx/utils/tests/test_decorators.py

@@ -1,510 +0,0 @@
-import os
-import pathlib
-import random
-import tempfile
-
-import pytest
-
-import networkx as nx
-from networkx.utils.decorators import (
-    argmap,
-    not_implemented_for,
-    np_random_state,
-    open_file,
-    py_random_state,
-)
-from networkx.utils.misc import PythonRandomInterface, PythonRandomViaNumpyBits
-
-
-def test_not_implemented_decorator():
-    @not_implemented_for("directed")
-    def test_d(G):
-        pass
-
-    test_d(nx.Graph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_d(nx.DiGraph())
-
-    @not_implemented_for("undirected")
-    def test_u(G):
-        pass
-
-    test_u(nx.DiGraph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_u(nx.Graph())
-
-    @not_implemented_for("multigraph")
-    def test_m(G):
-        pass
-
-    test_m(nx.Graph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_m(nx.MultiGraph())
-
-    @not_implemented_for("graph")
-    def test_g(G):
-        pass
-
-    test_g(nx.MultiGraph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_g(nx.Graph())
-
-    # not MultiDiGraph  (multiple arguments => AND)
-    @not_implemented_for("directed", "multigraph")
-    def test_not_md(G):
-        pass
-
-    test_not_md(nx.Graph())
-    test_not_md(nx.DiGraph())
-    test_not_md(nx.MultiGraph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_not_md(nx.MultiDiGraph())
-
-    # Graph only      (multiple decorators =>  OR)
-    @not_implemented_for("directed")
-    @not_implemented_for("multigraph")
-    def test_graph_only(G):
-        pass
-
-    test_graph_only(nx.Graph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_graph_only(nx.DiGraph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_graph_only(nx.MultiGraph())
-    with pytest.raises(nx.NetworkXNotImplemented):
-        test_graph_only(nx.MultiDiGraph())
-
-    with pytest.raises(ValueError):
-        not_implemented_for("directed", "undirected")
-
-    with pytest.raises(ValueError):
-        not_implemented_for("multigraph", "graph")
-
-
-def test_not_implemented_decorator_key():
-    with pytest.raises(KeyError):
-
-        @not_implemented_for("foo")
-        def test1(G):
-            pass
-
-        test1(nx.Graph())
-
-
-def test_not_implemented_decorator_raise():
-    with pytest.raises(nx.NetworkXNotImplemented):
-
-        @not_implemented_for("graph")
-        def test1(G):
-            pass
-
-        test1(nx.Graph())
-
-
-class TestOpenFileDecorator:
-    def setup_method(self):
-        self.text = ["Blah... ", "BLAH ", "BLAH!!!!"]
-        self.fobj = tempfile.NamedTemporaryFile("wb+", delete=False)
-        self.name = self.fobj.name
-
-    def teardown_method(self):
-        self.fobj.close()
-        os.unlink(self.name)
-
-    def write(self, path):
-        for text in self.text:
-            path.write(text.encode("ascii"))
-
-    @open_file(1, "r")
-    def read(self, path):
-        return path.readlines()[0]
-
-    @staticmethod
-    @open_file(0, "wb")
-    def writer_arg0(path):
-        path.write(b"demo")
-
-    @open_file(1, "wb+")
-    def writer_arg1(self, path):
-        self.write(path)
-
-    @open_file(2, "wb")
-    def writer_arg2default(self, x, path=None):
-        if path is None:
-            with tempfile.NamedTemporaryFile("wb+") as fh:
-                self.write(fh)
-        else:
-            self.write(path)
-
-    @open_file(4, "wb")
-    def writer_arg4default(self, x, y, other="hello", path=None, **kwargs):
-        if path is None:
-            with tempfile.NamedTemporaryFile("wb+") as fh:
-                self.write(fh)
-        else:
-            self.write(path)
-
-    @open_file("path", "wb")
-    def writer_kwarg(self, **kwargs):
-        path = kwargs.get("path", None)
-        if path is None:
-            with tempfile.NamedTemporaryFile("wb+") as fh:
-                self.write(fh)
-        else:
-            self.write(path)
-
-    def test_writer_arg0_str(self):
-        self.writer_arg0(self.name)
-
-    def test_writer_arg0_fobj(self):
-        self.writer_arg0(self.fobj)
-
-    def test_writer_arg0_pathlib(self):
-        self.writer_arg0(pathlib.Path(self.name))
-
-    def test_writer_arg1_str(self):
-        self.writer_arg1(self.name)
-        assert self.read(self.name) == "".join(self.text)
-
-    def test_writer_arg1_fobj(self):
-        self.writer_arg1(self.fobj)
-        assert not self.fobj.closed
-        self.fobj.close()
-        assert self.read(self.name) == "".join(self.text)
-
-    def test_writer_arg2default_str(self):
-        self.writer_arg2default(0, path=None)
-        self.writer_arg2default(0, path=self.name)
-        assert self.read(self.name) == "".join(self.text)
-
-    def test_writer_arg2default_fobj(self):
-        self.writer_arg2default(0, path=self.fobj)
-        assert not self.fobj.closed
-        self.fobj.close()
-        assert self.read(self.name) == "".join(self.text)
-
-    def test_writer_arg2default_fobj_path_none(self):
-        self.writer_arg2default(0, path=None)
-
-    def test_writer_arg4default_fobj(self):
-        self.writer_arg4default(0, 1, dog="dog", other="other")
-        self.writer_arg4default(0, 1, dog="dog", other="other", path=self.name)
-        assert self.read(self.name) == "".join(self.text)
-
-    def test_writer_kwarg_str(self):
-        self.writer_kwarg(path=self.name)
-        assert self.read(self.name) == "".join(self.text)
-
-    def test_writer_kwarg_fobj(self):
-        self.writer_kwarg(path=self.fobj)
-        self.fobj.close()
-        assert self.read(self.name) == "".join(self.text)
-
-    def test_writer_kwarg_path_none(self):
-        self.writer_kwarg(path=None)
-
-
-class TestRandomState:
-    @classmethod
-    def setup_class(cls):
-        global np
-        np = pytest.importorskip("numpy")
-
-    @np_random_state(1)
-    def instantiate_np_random_state(self, random_state):
-        allowed = (np.random.RandomState, np.random.Generator)
-        assert isinstance(random_state, allowed)
-        return random_state.random()
-
-    @py_random_state(1)
-    def instantiate_py_random_state(self, random_state):
-        allowed = (random.Random, PythonRandomInterface, PythonRandomViaNumpyBits)
-        assert isinstance(random_state, allowed)
-        return random_state.random()
-
-    def test_random_state_None(self):
-        np.random.seed(42)
-        rv = np.random.random()
-        np.random.seed(42)
-        assert rv == self.instantiate_np_random_state(None)
-
-        random.seed(42)
-        rv = random.random()
-        random.seed(42)
-        assert rv == self.instantiate_py_random_state(None)
-
-    def test_random_state_np_random(self):
-        np.random.seed(42)
-        rv = np.random.random()
-        np.random.seed(42)
-        assert rv == self.instantiate_np_random_state(np.random)
-        np.random.seed(42)
-        assert rv == self.instantiate_py_random_state(np.random)
-
-    def test_random_state_int(self):
-        np.random.seed(42)
-        np_rv = np.random.random()
-        random.seed(42)
-        py_rv = random.random()
-
-        np.random.seed(42)
-        seed = 1
-        rval = self.instantiate_np_random_state(seed)
-        rval_expected = np.random.RandomState(seed).rand()
-        assert rval == rval_expected
-        # test that global seed wasn't changed in function
-        assert np_rv == np.random.random()
-
-        random.seed(42)
-        rval = self.instantiate_py_random_state(seed)
-        rval_expected = random.Random(seed).random()
-        assert rval == rval_expected
-        # test that global seed wasn't changed in function
-        assert py_rv == random.random()
-
-    def test_random_state_np_random_Generator(self):
-        np.random.seed(42)
-        np_rv = np.random.random()
-        np.random.seed(42)
-        seed = 1
-
-        rng = np.random.default_rng(seed)
-        rval = self.instantiate_np_random_state(rng)
-        rval_expected = np.random.default_rng(seed).random()
-        assert rval == rval_expected
-
-        rval = self.instantiate_py_random_state(rng)
-        rval_expected = np.random.default_rng(seed).random(size=2)[1]
-        assert rval == rval_expected
-        # test that global seed wasn't changed in function
-        assert np_rv == np.random.random()
-
-    def test_random_state_np_random_RandomState(self):
-        np.random.seed(42)
-        np_rv = np.random.random()
-        np.random.seed(42)
-        seed = 1
-
-        rng = np.random.RandomState(seed)
-        rval = self.instantiate_np_random_state(rng)
-        rval_expected = np.random.RandomState(seed).random()
-        assert rval == rval_expected
-
-        rval = self.instantiate_py_random_state(rng)
-        rval_expected = np.random.RandomState(seed).random(size=2)[1]
-        assert rval == rval_expected
-        # test that global seed wasn't changed in function
-        assert np_rv == np.random.random()
-
-    def test_random_state_py_random(self):
-        seed = 1
-        rng = random.Random(seed)
-        rv = self.instantiate_py_random_state(rng)
-        assert rv == random.Random(seed).random()
-
-        pytest.raises(ValueError, self.instantiate_np_random_state, rng)
-
-
-def test_random_state_string_arg_index():
-    with pytest.raises(nx.NetworkXError):
-
-        @np_random_state("a")
-        def make_random_state(rs):
-            pass
-
-        rstate = make_random_state(1)
-
-
-def test_py_random_state_string_arg_index():
-    with pytest.raises(nx.NetworkXError):
-
-        @py_random_state("a")
-        def make_random_state(rs):
-            pass
-
-        rstate = make_random_state(1)
-
-
-def test_random_state_invalid_arg_index():
-    with pytest.raises(nx.NetworkXError):
-
-        @np_random_state(2)
-        def make_random_state(rs):
-            pass
-
-        rstate = make_random_state(1)
-
-
-def test_py_random_state_invalid_arg_index():
-    with pytest.raises(nx.NetworkXError):
-
-        @py_random_state(2)
-        def make_random_state(rs):
-            pass
-
-        rstate = make_random_state(1)
-
-
-class TestArgmap:
-    class ArgmapError(RuntimeError):
-        pass
-
-    def test_trivial_function(self):
-        def do_not_call(x):
-            raise ArgmapError("do not call this function")
-
-        @argmap(do_not_call)
-        def trivial_argmap():
-            return 1
-
-        assert trivial_argmap() == 1
-
-    def test_trivial_iterator(self):
-        def do_not_call(x):
-            raise ArgmapError("do not call this function")
-
-        @argmap(do_not_call)
-        def trivial_argmap():
-            yield from (1, 2, 3)
-
-        assert tuple(trivial_argmap()) == (1, 2, 3)
-
-    def test_contextmanager(self):
-        container = []
-
-        def contextmanager(x):
-            nonlocal container
-            return x, lambda: container.append(x)
-
-        @argmap(contextmanager, 0, 1, 2, try_finally=True)
-        def foo(x, y, z):
-            return x, y, z
-
-        x, y, z = foo("a", "b", "c")
-
-        # context exits are called in reverse
-        assert container == ["c", "b", "a"]
-
-    def test_tryfinally_generator(self):
-        container = []
-
-        def singleton(x):
-            return (x,)
-
-        with pytest.raises(nx.NetworkXError):
-
-            @argmap(singleton, 0, 1, 2, try_finally=True)
-            def foo(x, y, z):
-                yield from (x, y, z)
-
-        @argmap(singleton, 0, 1, 2)
-        def foo(x, y, z):
-            return x + y + z
-
-        q = foo("a", "b", "c")
-
-        assert q == ("a", "b", "c")
-
-    def test_actual_vararg(self):
-        @argmap(lambda x: -x, 4)
-        def foo(x, y, *args):
-            return (x, y) + tuple(args)
-
-        assert foo(1, 2, 3, 4, 5, 6) == (1, 2, 3, 4, -5, 6)
-
-    def test_signature_destroying_intermediate_decorator(self):
-        def add_one_to_first_bad_decorator(f):
-            """Bad because it doesn't wrap the f signature (clobbers it)"""
-
-            def decorated(a, *args, **kwargs):
-                return f(a + 1, *args, **kwargs)
-
-            return decorated
-
-        add_two_to_second = argmap(lambda b: b + 2, 1)
-
-        @add_two_to_second
-        @add_one_to_first_bad_decorator
-        def add_one_and_two(a, b):
-            return a, b
-
-        assert add_one_and_two(5, 5) == (6, 7)
-
-    def test_actual_kwarg(self):
-        @argmap(lambda x: -x, "arg")
-        def foo(*, arg):
-            return arg
-
-        assert foo(arg=3) == -3
-
-    def test_nested_tuple(self):
-        def xform(x, y):
-            u, v = y
-            return x + u + v, (x + u, x + v)
-
-        # we're testing args and kwargs here, too
-        @argmap(xform, (0, ("t", 2)))
-        def foo(a, *args, **kwargs):
-            return a, args, kwargs
-
-        a, args, kwargs = foo(1, 2, 3, t=4)
-
-        assert a == 1 + 4 + 3
-        assert args == (2, 1 + 3)
-        assert kwargs == {"t": 1 + 4}
-
-    def test_flatten(self):
-        assert tuple(argmap._flatten([[[[[], []], [], []], [], [], []]], set())) == ()
-
-        rlist = ["a", ["b", "c"], [["d"], "e"], "f"]
-        assert "".join(argmap._flatten(rlist, set())) == "abcdef"
-
-    def test_indent(self):
-        code = "\n".join(
-            argmap._indent(
-                *[
-                    "try:",
-                    "try:",
-                    "pass#",
-                    "finally:",
-                    "pass#",
-                    "#",
-                    "finally:",
-                    "pass#",
-                ]
-            )
-        )
-        assert (
-            code
-            == """try:
- try:
-  pass#
- finally:
-  pass#
- #
-finally:
- pass#"""
-        )
-
-    def test_immediate_raise(self):
-        @not_implemented_for("directed")
-        def yield_nodes(G):
-            yield from G
-
-        G = nx.Graph([(1, 2)])
-        D = nx.DiGraph()
-
-        # test first call (argmap is compiled and executed)
-        with pytest.raises(nx.NetworkXNotImplemented):
-            node_iter = yield_nodes(D)
-
-        # test second call (argmap is only executed)
-        with pytest.raises(nx.NetworkXNotImplemented):
-            node_iter = yield_nodes(D)
-
-        # ensure that generators still make generators
-        node_iter = yield_nodes(G)
-        next(node_iter)
-        next(node_iter)
-        with pytest.raises(StopIteration):
-            next(node_iter)

package/runtime/python/okstra_vendor/networkx/utils/tests/test_heaps.py

@@ -1,131 +0,0 @@
-import pytest
-
-import networkx as nx
-from networkx.utils import BinaryHeap, PairingHeap
-
-
-class X:
-    def __eq__(self, other):
-        raise self is other
-
-    def __ne__(self, other):
-        raise self is not other
-
-    def __lt__(self, other):
-        raise TypeError("cannot compare")
-
-    def __le__(self, other):
-        raise TypeError("cannot compare")
-
-    def __ge__(self, other):
-        raise TypeError("cannot compare")
-
-    def __gt__(self, other):
-        raise TypeError("cannot compare")
-
-    def __hash__(self):
-        return hash(id(self))
-
-
-x = X()
-
-
-data = [  # min should not invent an element.
-    ("min", nx.NetworkXError),
-    # Popping an empty heap should fail.
-    ("pop", nx.NetworkXError),
-    # Getting nonexisting elements should return None.
-    ("get", 0, None),
-    ("get", x, None),
-    ("get", None, None),
-    # Inserting a new key should succeed.
-    ("insert", x, 1, True),
-    ("get", x, 1),
-    ("min", (x, 1)),
-    # min should not pop the top element.
-    ("min", (x, 1)),
-    # Inserting a new key of different type should succeed.
-    ("insert", 1, -2.0, True),
-    # int and float values should interop.
-    ("min", (1, -2.0)),
-    # pop removes minimum-valued element.
-    ("insert", 3, -(10**100), True),
-    ("insert", 4, 5, True),
-    ("pop", (3, -(10**100))),
-    ("pop", (1, -2.0)),
-    # Decrease-insert should succeed.
-    ("insert", 4, -50, True),
-    ("insert", 4, -60, False, True),
-    # Decrease-insert should not create duplicate keys.
-    ("pop", (4, -60)),
-    ("pop", (x, 1)),
-    # Popping all elements should empty the heap.
-    ("min", nx.NetworkXError),
-    ("pop", nx.NetworkXError),
-    # Non-value-changing insert should fail.
-    ("insert", x, 0, True),
-    ("insert", x, 0, False, False),
-    ("min", (x, 0)),
-    ("insert", x, 0, True, False),
-    ("min", (x, 0)),
-    # Failed insert should not create duplicate keys.
-    ("pop", (x, 0)),
-    ("pop", nx.NetworkXError),
-    # Increase-insert should succeed when allowed.
-    ("insert", None, 0, True),
-    ("insert", 2, -1, True),
-    ("min", (2, -1)),
-    ("insert", 2, 1, True, False),
-    ("min", (None, 0)),
-    # Increase-insert should fail when disallowed.
-    ("insert", None, 2, False, False),
-    ("min", (None, 0)),
-    # Failed increase-insert should not create duplicate keys.
-    ("pop", (None, 0)),
-    ("pop", (2, 1)),
-    ("min", nx.NetworkXError),
-    ("pop", nx.NetworkXError),
-]
-
-
-def _test_heap_class(cls, *args, **kwargs):
-    heap = cls(*args, **kwargs)
-    # Basic behavioral test
-    for op in data:
-        if op[-1] is not nx.NetworkXError:
-            assert op[-1] == getattr(heap, op[0])(*op[1:-1])
-        else:
-            pytest.raises(op[-1], getattr(heap, op[0]), *op[1:-1])
-    # Coverage test.
-    for i in range(99, -1, -1):
-        assert heap.insert(i, i)
-    for i in range(50):
-        assert heap.pop() == (i, i)
-    for i in range(100):
-        assert heap.insert(i, i) == (i < 50)
-    for i in range(100):
-        assert not heap.insert(i, i + 1)
-    for i in range(50):
-        assert heap.pop() == (i, i)
-    for i in range(100):
-        assert heap.insert(i, i + 1) == (i < 50)
-    for i in range(49):
-        assert heap.pop() == (i, i + 1)
-    assert sorted([heap.pop(), heap.pop()]) == [(49, 50), (50, 50)]
-    for i in range(51, 100):
-        assert not heap.insert(i, i + 1, True)
-    for i in range(51, 70):
-        assert heap.pop() == (i, i + 1)
-    for i in range(100):
-        assert heap.insert(i, i)
-    for i in range(100):
-        assert heap.pop() == (i, i)
-    pytest.raises(nx.NetworkXError, heap.pop)
-
-
-def test_PairingHeap():
-    _test_heap_class(PairingHeap)
-
-
-def test_BinaryHeap():
-    _test_heap_class(BinaryHeap)