Skip to content

Commit 41fe232

Browse files
JulienChampagnolJulienChampagnol
committed
update unit tests
1 parent 61edda6 commit 41fe232

File tree

1 file changed

+72
-62
lines changed

1 file changed

+72
-62
lines changed

‎tests/test_functions.py‎

Lines changed: 72 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -51,71 +51,62 @@ def test_missing_files():
5151
assert type(additional_files) is list
5252

5353

54-
def test_geode_objects_output_extensions():
55-
for geode_object, value in geode_objects.geode_objects_dict().items():
56-
geode_objects_and_output_extensions_list = (
57-
geode_functions.geode_objects_output_extensions(geode_object)
58-
)
59-
assert type(geode_objects_and_output_extensions_list) is list
60-
for (
61-
geode_object_and_output_extensions
62-
) in geode_objects_and_output_extensions_list:
63-
assert type(geode_object_and_output_extensions) is dict
64-
assert "geode_object" in geode_object_and_output_extensions
65-
assert "output_extension" in geode_object_and_output_extensions
66-
67-
6854
def test_load():
6955
for geode_object, value in geode_objects.geode_objects_dict().items():
7056
print(f"\n{geode_object=}")
7157
input_extensions = geode_functions.geode_object_input_extensions(geode_object)
7258
for input_extension in input_extensions:
73-
# if input_extension not in ["grdecl", "og_img3d", "vti"]:
74-
print(f"\t{input_extension=}")
75-
missing_files = geode_functions.missing_files(
76-
geode_object, f"tests/data/test.{input_extension}"
77-
)
78-
has_missing_files = missing_files.has_missing_files()
79-
if has_missing_files:
80-
mandatory_files = missing_files.mandatory_files
81-
print(f"\t\t{mandatory_files=}")
82-
additional_files = missing_files.additional_files
83-
print(f"\t\t{additional_files=}")
84-
file_apsolute_path = os.path.abspath(f"tests/data/test.{input_extension}")
85-
86-
data = geode_functions.load(geode_object, file_apsolute_path)
87-
geode_functions.geode_objects_output_extensions(geode_object)
88-
for (
89-
geode_object_and_output_extensions
90-
) in geode_functions.geode_objects_output_extensions(geode_object):
91-
output_geode_object = geode_object_and_output_extensions["geode_object"]
92-
print(f"\t\t{output_geode_object=}")
93-
output_extensions = geode_functions.geode_object_output_extensions(
94-
geode_object
59+
if geode_object != "RegularGrid3D" and input_extension != "vti":
60+
print(f"\t{input_extension=}")
61+
missing_files = geode_functions.missing_files(
62+
geode_object, f"tests/data/test.{input_extension}"
63+
)
64+
has_missing_files = missing_files.has_missing_files()
65+
if has_missing_files:
66+
mandatory_files = missing_files.mandatory_files
67+
print(f"\t\t{mandatory_files=}")
68+
additional_files = missing_files.additional_files
69+
print(f"\t\t{additional_files=}")
70+
file_apsolute_path = os.path.abspath(
71+
f"tests/data/test.{input_extension}"
9572
)
9673

97-
for output_extension in output_extensions:
98-
print(f"\t\t\t{output_extension=}")
99-
uu_id = str(uuid.uuid4()).replace("-", "")
100-
filename = f"{uu_id}.{output_extension}"
101-
if geode_functions.is_saveable(geode_object, data, filename):
102-
file_path = geode_functions.save(
103-
geode_object,
104-
data,
105-
os.path.abspath(f"output"),
106-
f"{uu_id}.{output_extension}",
107-
)
108-
109-
if "save_viewable" in value:
110-
uu_id = str(uuid.uuid4()).replace("-", "")
111-
viewable_file_path = geode_functions.save_viewable(
74+
data = geode_functions.load(geode_object, file_apsolute_path)
75+
for (
76+
geode_object_and_output_extensions
77+
) in geode_functions.geode_objects_output_extensions(
78+
geode_object, data
79+
):
80+
output_geode_object = geode_object_and_output_extensions[
81+
"geode_object"
82+
]
83+
print(f"\t\t{output_geode_object=}")
84+
output_extensions = geode_functions.geode_object_output_extensions(
85+
geode_object
86+
)
87+
88+
for output_extension in output_extensions:
89+
print(f"\t\t\t{output_extension=}")
90+
uu_id = str(uuid.uuid4()).replace("-", "")
91+
filename = f"{uu_id}.{output_extension}"
92+
if geode_functions.is_saveable(geode_object, data, filename):
93+
file_path = geode_functions.save(
11294
geode_object,
11395
data,
11496
os.path.abspath(f"output"),
115-
uu_id,
97+
f"{uu_id}.{output_extension}",
11698
)
11799

118-
os.remove(viewable_file_path)
100+
if "save_viewable" in value:
101+
uu_id = str(uuid.uuid4()).replace("-", "")
102+
viewable_file_path = geode_functions.save_viewable(
103+
geode_object,
104+
data,
105+
os.path.abspath(f"output"),
106+
uu_id,
107+
)
108+
109+
os.remove(viewable_file_path)
119110

120111

121112
def test_is_model():
@@ -209,15 +200,34 @@ def test_list_geode_objects():
209200

210201
def test_geode_objects_output_extensions():
211202
for geode_object, value in geode_objects.geode_objects_dict().items():
212-
geode_objects_and_output_extensions_list = (
213-
geode_functions.geode_objects_output_extensions(geode_object)
214-
)
215-
print(f"{geode_objects_and_output_extensions_list=}")
216-
assert type(geode_objects_and_output_extensions_list) is list
217-
for (
218-
geode_object_and_output_extensions
219-
) in geode_objects_and_output_extensions_list:
220-
assert type(geode_object_and_output_extensions) is dict
203+
input_extensions = geode_functions.geode_object_input_extensions(geode_object)
204+
for input_extension in input_extensions:
205+
if input_extension not in ["og_img3d", "vti"]:
206+
data = geode_functions.load(
207+
geode_object, f"tests/data/test.{input_extension}"
208+
)
209+
geode_objects_and_output_extensions_list = (
210+
geode_functions.geode_objects_output_extensions(geode_object, data)
211+
)
212+
print(f"{geode_objects_and_output_extensions_list=}")
213+
assert type(geode_objects_and_output_extensions_list) is list
214+
for (
215+
geode_object_and_output_extensions
216+
) in geode_objects_and_output_extensions_list:
217+
print(f"{geode_object_and_output_extensions=}")
218+
assert type(geode_object_and_output_extensions) is dict
219+
assert "geode_object" in geode_object_and_output_extensions
220+
assert (
221+
type(geode_object_and_output_extensions["geode_object"]) is str
222+
)
223+
assert "outputs" in geode_object_and_output_extensions
224+
assert type(geode_object_and_output_extensions["outputs"]) is list
225+
outputs = geode_object_and_output_extensions["outputs"]
226+
for output in outputs:
227+
assert "extension" in output
228+
assert type(output["extension"]) is str
229+
assert "is_saveable" in output
230+
assert type(output["is_saveable"]) is bool
221231

222232

223233
def test_versions():

0 commit comments

Comments
 (0)