Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#!/usr/bin/env python
import contextlib
from dagflow.input import Input, Inputs
from dagflow.legs import Legs
from dagflow.output import Output
def test_01():
inputs = Inputs()
input1 = Input("i1", None)
input2 = Input("i2", None)
input3 = Input("i3", None)
inputs.add( (input1, input2) )
inputs.add( input3 )
print(inputs)
print(inputs[0])
print(inputs[1])
print(inputs[2])
try:
print(inputs[3])
except IndexError:
pass
else:
raise RuntimeError("fail")
print(inputs["i1"])
print(inputs["i2"])
print(inputs[("i1", "i3")])
print(inputs["i1"])
print(inputs["i2"])
print(inputs["i3"])
with contextlib.suppress(KeyError):
print(inputs["i4"])
def test_02():
inputs = Inputs()
print(inputs)
output1 = Output("o1", None)
try:
inputs.add( output1 )
except Exception:
pass
else:
raise RuntimeError("fail")
def test_03():
print("test3")
input1 = Input("i1", None)
input2 = Input("i2", None)
input3 = Input("i3", None)
output1 = Output("o1", None)
output2 = Output("o2", None)
legs = Legs((input1, input2, input3), (output1, output2))
print(legs)
legs.print()
print()
legs1 = legs[None, "o1"]
print(legs1)
# legs1.print()
print()
legs2 = legs[:, "o1"]
print(legs2)
legs2.print()
print()
legs3 = legs[("i1", "i3"), "o1"]
print(legs3)
legs3.print()
print()