Update tests

Signed-off-by: Laura Couto <laurarccouto@gmail.com>
kedro-org · Aug 8, 2024 · 4b2aac0 · 4b2aac0
1 parent cd5aa20
commit 4b2aac0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/framework/cli/test_cli.py b/tests/framework/cli/test_cli.py
@@ -520,7 +520,7 @@ def test_main_hook_exception_handling(self, fake_metadata):
             project_metadata=kedro_cli._metadata, command_args=[], exit_code=1
         )
 
-        assert "An error has occurred: Test Exception" in result.output
+        assert result.exit_code == 1
 
     @patch("sys.exit")
     def test_main_hook_finally_block(self, fake_metadata):
@@ -535,7 +535,7 @@ def test_main_hook_finally_block(self, fake_metadata):
             project_metadata=kedro_cli._metadata, command_args=[], exit_code=0
         )
 
-        assert "An error has occurred:" not in result.output
+        assert result.exit_code == 0
 
 
 @mark.usefixtures("chdir_to_dummy_project")