starting to test contexts ...

tests/core/test_contexts.py

@@ -0,0 +1,34 @@
+from bonobo import Graph
+from bonobo.core.contexts import ExecutionContext
+
+
+def generate_integers():
+    yield from range(10)
+
+
+def square(i: int) -> int:
+    return i**2
+
+
+def test_empty_execution_context():
+    graph = Graph()
+
+    ctx = ExecutionContext(graph)
+    assert not len(ctx.components)
+    assert not len(ctx.plugins)
+
+    assert not ctx.running
+
+
+def test_simple_execution_context():
+    graph = Graph()
+    graph.add_chain(generate_integers, square)
+
+    ctx = ExecutionContext(graph)
+    assert len(ctx.components) == 2
+    assert not len(ctx.plugins)
+
+    assert ctx[0].component is generate_integers
+    assert ctx[1].component is square
+
+    assert not ctx.running