|
19 | 19 | # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
20 | 20 | # SOFTWARE. |
21 | 21 |
|
22 | | -import os, sys, platform |
23 | | -if sys.version_info >= (3,8,0) and platform.system() == "Windows": |
| 22 | +import os |
| 23 | +import sys |
| 24 | +import platform |
| 25 | +if sys.version_info >= (3, 8, 0) and platform.system() == "Windows": |
24 | 26 | for path in [x.strip() for x in os.environ['PATH'].split(';') if x]: |
25 | 27 | os.add_dll_directory(path) |
26 | 28 |
|
27 | 29 | import opengeode_py_mesh as mesh |
28 | 30 |
|
29 | | -def test_create_vertices( graph, builder ): |
| 31 | + |
| 32 | +def test_create_vertices(graph, builder): |
30 | 33 | builder.create_vertex() |
31 | 34 | if graph.nb_vertices() != 1: |
32 | | - raise ValueError( "[Test] Graph should have 1 vertex" ) |
33 | | - builder.create_vertices( 3 ) |
| 35 | + raise ValueError("[Test] Graph should have 1 vertex") |
| 36 | + builder.create_vertices(3) |
34 | 37 | if graph.nb_vertices() != 4: |
35 | | - raise ValueError( "[Test] Graph should have 4 vertices" ) |
| 38 | + raise ValueError("[Test] Graph should have 4 vertices") |
36 | 39 |
|
37 | 40 |
|
38 | | -def test_create_edges( graph, builder ): |
39 | | - builder.create_edge_with_vertices( 0, 1 ) |
40 | | - builder.create_edge_with_vertices( 0, 2 ) |
41 | | - builder.create_edge_with_vertices( 3, 2 ) |
42 | | - builder.create_edge_with_vertices( 1, 2 ) |
| 41 | +def test_create_edges(graph, builder): |
| 42 | + builder.create_edge_with_vertices(0, 1) |
| 43 | + builder.create_edge_with_vertices(0, 2) |
| 44 | + builder.create_edge_with_vertices(3, 2) |
| 45 | + builder.create_edge_with_vertices(1, 2) |
43 | 46 | if graph.nb_edges() != 4: |
44 | | - raise ValueError( "[Test] Graph should have 4 edges" ) |
45 | | - answer = [ 3, 2 ] |
46 | | - if graph.edge_vertices( 2 ) != answer: |
47 | | - raise ValueError( "[Test] Wrong edge vertices" ) |
48 | | - |
49 | | - edges_around_0 = graph.edges_around_vertex( 0 ) |
50 | | - if len( edges_around_0 ) != 2: |
51 | | - raise ValueError( "[Test] edges_around_0 should have 2 edges" ) |
| 47 | + raise ValueError("[Test] Graph should have 4 edges") |
| 48 | + answer = [3, 2] |
| 49 | + if graph.edge_vertices(2) != answer: |
| 50 | + raise ValueError("[Test] Wrong edge vertices") |
| 51 | + |
| 52 | + edges_around_0 = graph.edges_around_vertex(0) |
| 53 | + if len(edges_around_0) != 2: |
| 54 | + raise ValueError("[Test] edges_around_0 should have 2 edges") |
52 | 55 | if edges_around_0[0].edge_id != 0: |
53 | | - raise ValueError( "[Test] edges_around_0 has wrong value" ) |
| 56 | + raise ValueError("[Test] edges_around_0 has wrong value") |
54 | 57 | if edges_around_0[0].vertex_id != 0: |
55 | | - raise ValueError( "[Test] edges_around_0 has wrong value" ) |
| 58 | + raise ValueError("[Test] edges_around_0 has wrong value") |
56 | 59 | if edges_around_0[1].edge_id != 1: |
57 | | - raise ValueError( "[Test] edges_around_0 has wrong value" ) |
| 60 | + raise ValueError("[Test] edges_around_0 has wrong value") |
58 | 61 | if edges_around_0[1].vertex_id != 0: |
59 | | - raise ValueError( "[Test] edges_around_0 has wrong value" ) |
| 62 | + raise ValueError("[Test] edges_around_0 has wrong value") |
60 | 63 |
|
61 | | - edges_around_2 = graph.edges_around_vertex( 2 ) |
62 | | - if len( edges_around_2 ) != 3: |
63 | | - raise ValueError( "[Test] edges_around_2 should have 3 edges" ) |
| 64 | + edges_around_2 = graph.edges_around_vertex(2) |
| 65 | + if len(edges_around_2) != 3: |
| 66 | + raise ValueError("[Test] edges_around_2 should have 3 edges") |
64 | 67 | if edges_around_2[0].edge_id != 1: |
65 | | - raise ValueError( "[Test] edges_around_2 has wrong value" ) |
| 68 | + raise ValueError("[Test] edges_around_2 has wrong value") |
66 | 69 | if edges_around_2[0].vertex_id != 1: |
67 | | - raise ValueError( "[Test] edges_around_2 has wrong value" ) |
| 70 | + raise ValueError("[Test] edges_around_2 has wrong value") |
68 | 71 | if edges_around_2[1].edge_id != 2: |
69 | | - raise ValueError( "[Test] edges_around_2 has wrong value" ) |
| 72 | + raise ValueError("[Test] edges_around_2 has wrong value") |
70 | 73 | if edges_around_2[1].vertex_id != 1: |
71 | | - raise ValueError( "[Test] edges_around_2 has wrong value" ) |
| 74 | + raise ValueError("[Test] edges_around_2 has wrong value") |
72 | 75 | if edges_around_2[2].edge_id != 3: |
73 | | - raise ValueError( "[Test] edges_around_2 has wrong value" ) |
| 76 | + raise ValueError("[Test] edges_around_2 has wrong value") |
74 | 77 | if edges_around_2[2].vertex_id != 1: |
75 | | - raise ValueError( "[Test] edges_around_2 has wrong value" ) |
| 78 | + raise ValueError("[Test] edges_around_2 has wrong value") |
| 79 | + |
76 | 80 |
|
77 | | -def test_delete_edge( graph, builder ): |
| 81 | +def test_delete_edge(graph, builder): |
78 | 82 | to_delete = [False] * graph.nb_edges() |
79 | 83 | to_delete[0] = True |
80 | | - builder.delete_edges( to_delete ) |
| 84 | + builder.delete_edges(to_delete) |
81 | 85 | if graph.nb_edges() != 3: |
82 | | - raise ValueError( "[Test] Graph should have 3 edges" ) |
83 | | - if graph.edge_vertex( mesh.EdgeVertex( 0, 0 ) ) != 0: |
84 | | - raise ValueError( "[Test] Graph edge vertex index is not correct" ) |
85 | | - if graph.edge_vertex( mesh.EdgeVertex( 0, 1 ) ) != 2: |
86 | | - raise ValueError( "[Test] Graph edge vertex index is not correct" ) |
87 | | - |
88 | | - builder.create_edges( 10 ) |
89 | | - builder.set_edge_vertex( mesh.EdgeVertex( 1, 0 ), 1 ) |
90 | | - builder.set_edge_vertex( mesh.EdgeVertex( 1, 1 ), 0 ) |
| 86 | + raise ValueError("[Test] Graph should have 3 edges") |
| 87 | + if graph.edge_vertex(mesh.EdgeVertex(0, 0)) != 0: |
| 88 | + raise ValueError("[Test] Graph edge vertex index is not correct") |
| 89 | + if graph.edge_vertex(mesh.EdgeVertex(0, 1)) != 2: |
| 90 | + raise ValueError("[Test] Graph edge vertex index is not correct") |
| 91 | + |
| 92 | + builder.create_edges(10) |
| 93 | + builder.set_edge_vertex(mesh.EdgeVertex(1, 0), 1) |
| 94 | + builder.set_edge_vertex(mesh.EdgeVertex(1, 1), 0) |
91 | 95 | if graph.nb_edges() != 13: |
92 | | - raise ValueError( "[Test] Graph should have 13 edges" ) |
| 96 | + raise ValueError("[Test] Graph should have 13 edges") |
93 | 97 |
|
94 | 98 | to_delete[-1] = True |
95 | | - to_delete.extend( [True] * 9 ) |
96 | | - builder.delete_edges( to_delete ) |
| 99 | + to_delete.extend([True] * 9) |
| 100 | + builder.delete_edges(to_delete) |
97 | 101 | if graph.nb_edges() != 2: |
98 | | - raise ValueError( "[Test] Graph should have 2 edges" ) |
99 | | - if graph.edge_vertex( mesh.EdgeVertex( 0, 0 ) ) != 1: |
100 | | - raise ValueError( "[Test] Graph edge vertex index is not correct (0, 0)" ) |
101 | | - if graph.edge_vertex( mesh.EdgeVertex( 0, 1 ) ) != 0: |
102 | | - raise ValueError( "[Test] Graph edge vertex index is not correct (0, 1)" ) |
| 102 | + raise ValueError("[Test] Graph should have 2 edges") |
| 103 | + if graph.edge_vertex(mesh.EdgeVertex(0, 0)) != 1: |
| 104 | + raise ValueError( |
| 105 | + "[Test] Graph edge vertex index is not correct (0, 0)") |
| 106 | + if graph.edge_vertex(mesh.EdgeVertex(0, 1)) != 0: |
| 107 | + raise ValueError( |
| 108 | + "[Test] Graph edge vertex index is not correct (0, 1)") |
| 109 | + |
103 | 110 |
|
104 | | -def test_io( graph, filename ): |
105 | | - mesh.save_graph( graph, filename ) |
106 | | - new_graph = mesh.load_graph( filename ) |
| 111 | +def test_io(graph, filename): |
| 112 | + mesh.save_graph(graph, filename) |
| 113 | + new_graph = mesh.load_graph(filename) |
107 | 114 |
|
108 | | -def test_clone( graph ): |
| 115 | + |
| 116 | +def test_clone(graph): |
109 | 117 | graph2 = graph.clone() |
110 | 118 | if graph2.nb_vertices() != 4: |
111 | | - raise ValueError( "[Test] Graph2 should have 4 vertices" ) |
| 119 | + raise ValueError("[Test] Graph2 should have 4 vertices") |
112 | 120 | if graph2.nb_edges() != 2: |
113 | | - raise ValueError( "[Test] Graph2 should have 2 edges" ) |
| 121 | + raise ValueError("[Test] Graph2 should have 2 edges") |
| 122 | + |
114 | 123 |
|
115 | | -def test_delete_isolated_vertices( graph, builder ): |
| 124 | +def test_delete_isolated_vertices(graph, builder): |
116 | 125 | builder.delete_isolated_vertices() |
117 | 126 | if graph.nb_vertices() != 3: |
118 | | - raise ValueError( "[Test] Graph should have 3 vertices" ) |
| 127 | + raise ValueError("[Test] Graph should have 3 vertices") |
119 | 128 | if graph.nb_edges() != 2: |
120 | | - raise ValueError( "[Test] Graph2 should have 2 edges" ) |
| 129 | + raise ValueError("[Test] Graph2 should have 2 edges") |
| 130 | + |
121 | 131 |
|
122 | 132 | if __name__ == '__main__': |
| 133 | + mesh.OpenGeodeMesh.initialize() |
123 | 134 | graph = mesh.Graph.create() |
124 | | - builder = mesh.GraphBuilder.create( graph ) |
125 | | - |
126 | | - test_create_vertices( graph, builder ) |
127 | | - test_create_edges( graph, builder ) |
128 | | - test_io( graph, "test." + graph.native_extension() ) |
129 | | - |
130 | | - test_delete_edge( graph, builder ) |
131 | | - test_clone( graph ) |
132 | | - test_delete_isolated_vertices( graph, builder ) |
| 135 | + builder = mesh.GraphBuilder.create(graph) |
| 136 | + |
| 137 | + test_create_vertices(graph, builder) |
| 138 | + test_create_edges(graph, builder) |
| 139 | + test_io(graph, "test." + graph.native_extension()) |
| 140 | + |
| 141 | + test_delete_edge(graph, builder) |
| 142 | + test_clone(graph) |
| 143 | + test_delete_isolated_vertices(graph, builder) |
0 commit comments