comparison env/lib/python3.9/site-packages/networkx/classes/tests/test_filters.py @ 0:4f3585e2f14b draft default tip

"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author shellac
date Mon, 22 Mar 2021 18:12:50 +0000
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:4f3585e2f14b
1 import pytest
2 import networkx as nx
3
4
5 class TestFilterFactory:
6 def test_no_filter(self):
7 nf = nx.filters.no_filter
8 assert nf()
9 assert nf(1)
10 assert nf(2, 1)
11
12 def test_hide_nodes(self):
13 f = nx.classes.filters.hide_nodes([1, 2, 3])
14 assert not f(1)
15 assert not f(2)
16 assert not f(3)
17 assert f(4)
18 assert f(0)
19 assert f("a")
20 pytest.raises(TypeError, f, 1, 2)
21 pytest.raises(TypeError, f)
22
23 def test_show_nodes(self):
24 f = nx.classes.filters.show_nodes([1, 2, 3])
25 assert f(1)
26 assert f(2)
27 assert f(3)
28 assert not f(4)
29 assert not f(0)
30 assert not f("a")
31 pytest.raises(TypeError, f, 1, 2)
32 pytest.raises(TypeError, f)
33
34 def test_hide_edges(self):
35 factory = nx.classes.filters.hide_edges
36 f = factory([(1, 2), (3, 4)])
37 assert not f(1, 2)
38 assert not f(3, 4)
39 assert not f(4, 3)
40 assert f(2, 3)
41 assert f(0, -1)
42 assert f("a", "b")
43 pytest.raises(TypeError, f, 1, 2, 3)
44 pytest.raises(TypeError, f, 1)
45 pytest.raises(TypeError, f)
46 pytest.raises(TypeError, factory, [1, 2, 3])
47 pytest.raises(ValueError, factory, [(1, 2, 3)])
48
49 def test_show_edges(self):
50 factory = nx.classes.filters.show_edges
51 f = factory([(1, 2), (3, 4)])
52 assert f(1, 2)
53 assert f(3, 4)
54 assert f(4, 3)
55 assert not f(2, 3)
56 assert not f(0, -1)
57 assert not f("a", "b")
58 pytest.raises(TypeError, f, 1, 2, 3)
59 pytest.raises(TypeError, f, 1)
60 pytest.raises(TypeError, f)
61 pytest.raises(TypeError, factory, [1, 2, 3])
62 pytest.raises(ValueError, factory, [(1, 2, 3)])
63
64 def test_hide_diedges(self):
65 factory = nx.classes.filters.hide_diedges
66 f = factory([(1, 2), (3, 4)])
67 assert not f(1, 2)
68 assert not f(3, 4)
69 assert f(4, 3)
70 assert f(2, 3)
71 assert f(0, -1)
72 assert f("a", "b")
73 pytest.raises(TypeError, f, 1, 2, 3)
74 pytest.raises(TypeError, f, 1)
75 pytest.raises(TypeError, f)
76 pytest.raises(TypeError, factory, [1, 2, 3])
77 pytest.raises(ValueError, factory, [(1, 2, 3)])
78
79 def test_show_diedges(self):
80 factory = nx.classes.filters.show_diedges
81 f = factory([(1, 2), (3, 4)])
82 assert f(1, 2)
83 assert f(3, 4)
84 assert not f(4, 3)
85 assert not f(2, 3)
86 assert not f(0, -1)
87 assert not f("a", "b")
88 pytest.raises(TypeError, f, 1, 2, 3)
89 pytest.raises(TypeError, f, 1)
90 pytest.raises(TypeError, f)
91 pytest.raises(TypeError, factory, [1, 2, 3])
92 pytest.raises(ValueError, factory, [(1, 2, 3)])
93
94 def test_hide_multiedges(self):
95 factory = nx.classes.filters.hide_multiedges
96 f = factory([(1, 2, 0), (3, 4, 1), (1, 2, 1)])
97 assert not f(1, 2, 0)
98 assert not f(1, 2, 1)
99 assert f(1, 2, 2)
100 assert f(3, 4, 0)
101 assert not f(3, 4, 1)
102 assert not f(4, 3, 1)
103 assert f(4, 3, 0)
104 assert f(2, 3, 0)
105 assert f(0, -1, 0)
106 assert f("a", "b", 0)
107 pytest.raises(TypeError, f, 1, 2, 3, 4)
108 pytest.raises(TypeError, f, 1, 2)
109 pytest.raises(TypeError, f, 1)
110 pytest.raises(TypeError, f)
111 pytest.raises(TypeError, factory, [1, 2, 3])
112 pytest.raises(ValueError, factory, [(1, 2)])
113 pytest.raises(ValueError, factory, [(1, 2, 3, 4)])
114
115 def test_show_multiedges(self):
116 factory = nx.classes.filters.show_multiedges
117 f = factory([(1, 2, 0), (3, 4, 1), (1, 2, 1)])
118 assert f(1, 2, 0)
119 assert f(1, 2, 1)
120 assert not f(1, 2, 2)
121 assert not f(3, 4, 0)
122 assert f(3, 4, 1)
123 assert f(4, 3, 1)
124 assert not f(4, 3, 0)
125 assert not f(2, 3, 0)
126 assert not f(0, -1, 0)
127 assert not f("a", "b", 0)
128 pytest.raises(TypeError, f, 1, 2, 3, 4)
129 pytest.raises(TypeError, f, 1, 2)
130 pytest.raises(TypeError, f, 1)
131 pytest.raises(TypeError, f)
132 pytest.raises(TypeError, factory, [1, 2, 3])
133 pytest.raises(ValueError, factory, [(1, 2)])
134 pytest.raises(ValueError, factory, [(1, 2, 3, 4)])
135
136 def test_hide_multidiedges(self):
137 factory = nx.classes.filters.hide_multidiedges
138 f = factory([(1, 2, 0), (3, 4, 1), (1, 2, 1)])
139 assert not f(1, 2, 0)
140 assert not f(1, 2, 1)
141 assert f(1, 2, 2)
142 assert f(3, 4, 0)
143 assert not f(3, 4, 1)
144 assert f(4, 3, 1)
145 assert f(4, 3, 0)
146 assert f(2, 3, 0)
147 assert f(0, -1, 0)
148 assert f("a", "b", 0)
149 pytest.raises(TypeError, f, 1, 2, 3, 4)
150 pytest.raises(TypeError, f, 1, 2)
151 pytest.raises(TypeError, f, 1)
152 pytest.raises(TypeError, f)
153 pytest.raises(TypeError, factory, [1, 2, 3])
154 pytest.raises(ValueError, factory, [(1, 2)])
155 pytest.raises(ValueError, factory, [(1, 2, 3, 4)])
156
157 def test_show_multidiedges(self):
158 factory = nx.classes.filters.show_multidiedges
159 f = factory([(1, 2, 0), (3, 4, 1), (1, 2, 1)])
160 assert f(1, 2, 0)
161 assert f(1, 2, 1)
162 assert not f(1, 2, 2)
163 assert not f(3, 4, 0)
164 assert f(3, 4, 1)
165 assert not f(4, 3, 1)
166 assert not f(4, 3, 0)
167 assert not f(2, 3, 0)
168 assert not f(0, -1, 0)
169 assert not f("a", "b", 0)
170 pytest.raises(TypeError, f, 1, 2, 3, 4)
171 pytest.raises(TypeError, f, 1, 2)
172 pytest.raises(TypeError, f, 1)
173 pytest.raises(TypeError, f)
174 pytest.raises(TypeError, factory, [1, 2, 3])
175 pytest.raises(ValueError, factory, [(1, 2)])
176 pytest.raises(ValueError, factory, [(1, 2, 3, 4)])