diff --git a/tests/test_train.py b/tests/test_train.py index 7c4bc48..a03cf6e 100644 --- a/tests/test_train.py +++ b/tests/test_train.py @@ -101,6 +101,7 @@ def test_train(): '-p', 'useless_config.json', '-d', path ]) + raise IOError(result.stdout) assert result.exit_code == 0 # assert that model checkpoint was created. assert os.path.exists(join(path, 'model', 'model.ckpt-10.index'))