From 55a628b3d02ebcb485ff55c29cfd56521546be91 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=9F=D0=B0=D0=B2=D0=B5=D0=BB=20=D0=96=D1=83=D0=BA=D0=BE?=
 =?UTF-8?q?=D0=B2?= <33721692+LeaveMyYard@users.noreply.github.com>
Date: Wed, 29 Mar 2023 18:12:02 +0300
Subject: Fix case when can not possible to calculate

---
 tests/test_krr.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

(limited to 'tests')

diff --git a/tests/test_krr.py b/tests/test_krr.py
index 4a25db6..30704ba 100644
--- a/tests/test_krr.py
+++ b/tests/test_krr.py
@@ -18,22 +18,31 @@ STRATEGY_NAME = "simple"
 
 def test_help():
     result = runner.invoke(app, [STRATEGY_NAME, "--help"])
-    assert result.exit_code == 0
+    try:
+        assert result.exit_code == 0
+    except AssertionError as e:
+        raise e from result.exception
 
 
 @pytest.mark.parametrize("log_flag", ["-v", "-q"])
 def test_run(log_flag: str):
     result = runner.invoke(app, [STRATEGY_NAME, log_flag, "--namespace", "default"])
-    assert result.exit_code == 0
+    try:
+        assert result.exit_code == 0
+    except AssertionError as e:
+        raise e from result.exception
 
 
 @pytest.mark.parametrize("format", ["json", "yaml", "table", "pprint"])
-def test_output_json(format: str):
+def test_output_formats(format: str):
     result = runner.invoke(app, [STRATEGY_NAME, "-q", "-f", format, "--namespace", "default"])
-    assert result.exit_code == 0
+    try:
+        assert result.exit_code == 0
+    except AssertionError as e:
+        raise e from result.exception
 
     if format == "json":
-        assert json.loads(result.stdout)
+        assert json.loads(result.stdout), result.stdout
 
     if format == "yaml":
-        assert yaml.safe_load(result.stdout)
+        assert yaml.safe_load(result.stdout), result.stdout
-- 
cgit v1.2.3