Skip to content

Commit 76f61f5

Browse files
authored
Merge branch 'main' into choose-grid
2 parents eb18888 + cb901e8 commit 76f61f5

1 file changed

Lines changed: 57 additions & 58 deletions

File tree

tests/test_morphio.py

Lines changed: 57 additions & 58 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def setup(self):
118118
self.testfiles.append(testsequence_dir.joinpath(filename))
119119
return
120120

121-
def test_morph_outputs(self, setup, tmp_path):
121+
def test_morph_outputs(self, setup, tmp_path):
122122
morph_file = self.testfiles[0]
123123
target_file = self.testfiles[-1]
124124

@@ -201,61 +201,60 @@ def test_morph_outputs(self, setup, tmp_path):
201201
expected = filter(ignore_path, tf)
202202
are_files_same(actual, expected)
203203

204-
# Save multiple verbose morphs with uncertainties
205-
tmp_verbose_unc = tmp_path.joinpath("verbose_unc")
206-
tmp_verbose_unc_name = tmp_verbose_unc.resolve().as_posix()
207-
208-
opts, pargs = self.parser.parse_args(
209-
[
210-
"--multiple-targets",
211-
"--sort-by",
212-
"temperature",
213-
"-s",
214-
tmp_verbose_unc_name,
215-
"-n",
216-
"--save-names-file",
217-
tssf,
218-
"--scale",
219-
"1",
220-
"--squeeze",
221-
"0,0",
222-
"-u",
223-
"--verbose",
224-
]
225-
)
226-
pargs = [morph_file, testsequence_dir]
227-
multiple_targets(self.parser, opts, pargs, stdout_flag=False)
228-
229-
# Save a single verbose morph
230-
svum = tmp_verbose_unc.joinpath("single_verbose_unc_morph.cgr")
231-
svum_name = svum.resolve().as_posix()
232-
opts, pargs = self.parser.parse_args(
233-
[
234-
"-s",
235-
svum_name,
236-
"-n",
237-
"--scale",
238-
"1",
239-
"--squeeze",
240-
"0,0",
241-
"-u",
242-
"--verbose",
243-
]
244-
)
245-
pargs = [morph_file, target_file]
246-
single_morph(self.parser, opts, pargs, stdout_flag=False)
247-
248-
# Check the saved files are the same for verbose
249-
common = []
250-
for item in tmp_verbose_unc.glob("**/*.*"):
251-
if item.is_file():
252-
common.append(item.relative_to(tmp_verbose_unc).as_posix())
253-
for file in common:
254-
with open(tmp_verbose_unc.joinpath(file)) as gf:
255-
with open(test_saving_verbose_unc.joinpath(file)) as tf:
256-
actual = filter(ignore_path, gf)
257-
expected = filter(ignore_path, tf)
258-
are_files_same(actual, expected)
204+
# Save multiple verbose morphs with uncertainties
205+
tmp_verbose_unc = tmp_path.joinpath("verbose_unc")
206+
tmp_verbose_unc_name = tmp_verbose_unc.resolve().as_posix()
207+
208+
opts, pargs = self.parser.parse_args(
209+
[
210+
"--multiple-targets",
211+
"--sort-by",
212+
"temperature",
213+
"-s",
214+
tmp_verbose_unc_name,
215+
"-n",
216+
"--save-names-file",
217+
tssf,
218+
"--scale",
219+
"1",
220+
"--squeeze",
221+
"0,0",
222+
"-u",
223+
"--verbose",
224+
]
225+
)
226+
pargs = [morph_file, testsequence_dir]
227+
multiple_targets(self.parser, opts, pargs, stdout_flag=False)
228+
229+
# Save a single verbose morph with un certainties
230+
svum = tmp_verbose_unc.joinpath("single_verbose_unc_morph.cgr")
231+
svum_name = svum.resolve().as_posix()
232+
opts, pargs = self.parser.parse_args(
233+
[
234+
"-s",
235+
svum_name,
236+
"-n",
237+
"--scale",
238+
"1",
239+
"--squeeze",
240+
"0,0",
241+
"-u",
242+
"--verbose",
243+
]
244+
)
245+
pargs = [morph_file, target_file]
246+
single_morph(self.parser, opts, pargs, stdout_flag=False)
247+
248+
common = []
249+
for item in tmp_verbose_unc.glob("**/*.*"):
250+
if item.is_file():
251+
common.append(item.relative_to(tmp_verbose_unc).as_posix())
252+
for file in common:
253+
with open(tmp_verbose_unc.joinpath(file)) as gf:
254+
with open(test_saving_verbose_unc.joinpath(file)) as tf:
255+
actual = filter(ignore_path, gf)
256+
expected = filter(ignore_path, tf)
257+
are_files_same(actual, expected)
259258

260259
def test_grid_selection(self, setup, tmp_path):
261260
morph_file = self.testfiles[0]
@@ -282,7 +281,7 @@ def test_grid_selection(self, setup, tmp_path):
282281

283282
tmp_default_morphs = tmp_default.joinpath("Morphs")
284283

285-
# Check the saved files are the same for verbose
284+
# Check the saved files are the same for default grid
286285
common = []
287286
for item in tmp_default_morphs.glob("**/*.*"):
288287
if item.is_file():
@@ -317,7 +316,7 @@ def test_grid_selection(self, setup, tmp_path):
317316

318317
tmp_original_morphs = tmp_default.joinpath("Morphs")
319318

320-
# Check the saved files are the same for verbose
319+
# Check the saved files are the same for original grid
321320
common = []
322321
for item in tmp_original_morphs.glob("**/*.*"):
323322
if item.is_file():

0 commit comments

Comments
 (0)