Skip to content

Commit 83c57ad

Browse files
xsaschakojubicker
andauthored
1300 Update surragete model test to work with new pyfakefs version (#1301)
Co-authored-by: jubicker <113909589+jubicker@users.noreply.github.com>
1 parent b6f044d commit 83c57ad

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

pycode/memilio-surrogatemodel/memilio/surrogatemodel_test/test_surrogatemodel_ode_secir_groups.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -505,9 +505,14 @@ def test_network_fit(self, mock_population, mock_baseline, mock_minimum):
505505
model.network_fit(
506506
path=self.path, model=md,
507507
modeltype='classic', training_parameter=training_parameter, filename="data_secir_groups.pickle")
508-
error_message = "[Errno 2] No such file or directory in the fake filesystem: '" + \
509-
self.path + "data_secir_groups.pickle'"
510-
self.assertEqual(str(error.exception), error_message)
508+
# check error message
509+
error_message_part1 = "[Errno 2] No such file or directory"
510+
error_message_part2 = self.path
511+
error_message_part3 = "data_secir_groups.pickle"
512+
513+
self.assertIn(error_message_part1, str(error.exception))
514+
self.assertIn(error_message_part2, str(error.exception))
515+
self.assertIn(error_message_part3, str(error.exception))
511516

512517
# generate dataset with multiple output
513518
input_width = 5

0 commit comments

Comments
 (0)