@@ -84,43 +84,45 @@ def test_load():
8484 )
8585 if is_loadable :
8686 data = geode_functions .load (geode_object , file_apsolute_path )
87+ if "save_viewable" in value :
88+ uu_id = str (uuid .uuid4 ()).replace ("-" , "" )
89+ viewable_file_path = geode_functions .save_viewable (
90+ geode_object ,
91+ data ,
92+ os .path .abspath (f"/output" ),
93+ uu_id ,
94+ )
95+ os .remove (viewable_file_path )
96+ geode_objects_and_output_extensions = (
97+ geode_functions .geode_objects_output_extensions (
98+ geode_object , data
99+ )
100+ )
101+ assert type (geode_objects_and_output_extensions ) is dict
87102 for (
88- geode_object_and_output_extensions
89- ) in geode_functions .geode_objects_output_extensions (
90- geode_object , data
91- ):
92- output_geode_object = geode_object_and_output_extensions [
93- "geode_object"
94- ]
103+ output_geode_object ,
104+ output_geode_object_value ,
105+ ) in geode_objects_and_output_extensions .items ():
95106 print (f"\t \t { output_geode_object = } " )
96- output_extensions = (
97- geode_functions .geode_object_output_extensions (geode_object )
98- )
99-
100- for output_extension in output_extensions :
107+ for (
108+ output_extension ,
109+ output_extension_value ,
110+ ) in output_geode_object_value .items ():
101111 print (f"\t \t \t { output_extension = } " )
102112 uu_id = str (uuid .uuid4 ()).replace ("-" , "" )
103113 filename = f"{ uu_id } .{ output_extension } "
104114 if geode_functions .is_saveable (
105- geode_object , data , filename
115+ output_geode_object , data , filename
106116 ):
107117 saved_files = geode_functions .save (
108- geode_object ,
118+ output_geode_object ,
109119 data ,
110- os .path .abspath (f"output" ),
111- f" { uu_id } . { output_extension } " ,
120+ os .path .abspath (f"/ output" ),
121+ filename ,
112122 )
113-
114- if "save_viewable" in value :
115- uu_id = str (uuid .uuid4 ()).replace ("-" , "" )
116- viewable_file_path = geode_functions .save_viewable (
117- geode_object ,
118- data ,
119- os .path .abspath (f"output" ),
120- uu_id ,
121- )
122-
123- os .remove (viewable_file_path )
123+ assert type (saved_files ) is list
124+ for saved_file in saved_files :
125+ os .remove (saved_file )
124126
125127
126128def test_is_model ():
@@ -151,10 +153,39 @@ def test_geode_object_input_extensions():
151153
152154def test_geode_object_output_extensions ():
153155 for geode_object , value in geode_objects .geode_objects_dict ().items ():
154- output_extensions = geode_functions .geode_object_output_extensions (geode_object )
155- assert type (output_extensions ) is list
156- for output_extension in output_extensions :
157- assert type (output_extension ) is str
156+ print (f"\n { geode_object = } " )
157+ input_extensions = geode_functions .geode_object_input_extensions (geode_object )
158+ for input_extension in input_extensions :
159+ if geode_object != "RegularGrid3D" and input_extension != "vti" :
160+ print (f"\t { input_extension = } " )
161+ missing_files = geode_functions .missing_files (
162+ geode_object , f"tests/data/test.{ input_extension } "
163+ )
164+ has_missing_files = missing_files .has_missing_files ()
165+ if has_missing_files :
166+ mandatory_files = missing_files .mandatory_files
167+ print (f"\t \t { mandatory_files = } " )
168+ additional_files = missing_files .additional_files
169+ print (f"\t \t { additional_files = } " )
170+ file_apsolute_path = os .path .abspath (
171+ f"tests/data/test.{ input_extension } "
172+ )
173+
174+ data = geode_functions .load (geode_object , file_apsolute_path )
175+ geode_objets_and_output_extensions = (
176+ geode_functions .geode_objects_output_extensions (geode_object , data )
177+ )
178+ assert type (geode_objets_and_output_extensions ) is dict
179+ for (
180+ output_geode_object ,
181+ output_geode_object_value ,
182+ ) in geode_objets_and_output_extensions .items ():
183+ for (
184+ output_extension ,
185+ output_extension_value ,
186+ ) in output_geode_object_value .items ():
187+ assert type (output_extension ) is str
188+ assert type (output_extension_value ["is_saveable" ]) is bool
158189
159190
160191def test_list_input_extensions ():
@@ -182,8 +213,6 @@ def test_list_geode_objects():
182213 "Graph" ,
183214 "RasterImage2D" ,
184215 "RasterImage3D" ,
185- "RasterImage2D" ,
186- "RasterImage3D" ,
187216 "VertexSet" ,
188217 ],
189218 },
@@ -216,34 +245,23 @@ def test_geode_objects_output_extensions():
216245 for geode_object , value in geode_objects .geode_objects_dict ().items ():
217246 input_extensions = geode_functions .geode_object_input_extensions (geode_object )
218247 for input_extension in input_extensions :
219- if input_extension not in [ "og_img3d" , "vti" ] :
248+ if geode_object != "RegularGrid3D" and input_extension != "vti" :
220249 data = geode_functions .load (
221250 geode_object , f"tests/data/test.{ input_extension } "
222251 )
223252 geode_objects_and_output_extensions = (
224- geode_functions .geode_objects_and_output_extensions (
225- geode_object , data
226- )
253+ geode_functions .geode_objects_output_extensions (geode_object , data )
227254 )
228- print (f"{ geode_objects_and_output_extensions = } " )
229255 assert type (geode_objects_and_output_extensions ) is dict
230256 for (
231- geode_object_and_output_extensions
232- ) in geode_objects_and_output_extensions .values ():
233- print (f"{ geode_object_and_output_extensions = } " )
234- assert type (geode_object_and_output_extensions ) is dict
235- assert "geode_object" in geode_object_and_output_extensions
236- assert (
237- type (geode_object_and_output_extensions ["geode_object" ]) is str
238- )
239- assert "outputs" in geode_object_and_output_extensions
240- assert type (geode_object_and_output_extensions ["outputs" ]) is list
241- outputs = geode_object_and_output_extensions ["outputs" ]
242- for output in outputs :
243- assert "extension" in output
244- assert type (output ["extension" ]) is str
245- assert "is_saveable" in output
246- assert type (output ["is_saveable" ]) is bool
257+ output_geode_object ,
258+ output_geode_object_value ,
259+ ) in geode_objects_and_output_extensions .items ():
260+ for (
261+ output_extension ,
262+ output_extension_value ,
263+ ) in output_geode_object_value .items ():
264+ assert type (output_extension_value ["is_saveable" ]) is bool
247265
248266
249267def test_versions ():
0 commit comments