Merge pull request #87 from qiboteam/format_err

Coloring the `JobApiError` in red
qiboteam · Nov 12, 2024 · 419ee5c · 419ee5c
2 parents 99609be + 0b3119d
commit 419ee5c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/src/qibo_client/exceptions.py b/src/qibo_client/exceptions.py
@@ -27,5 +27,5 @@ class JobApiError(Exception):
     def __init__(self, status_code: int, message: str):
         self.status_code = status_code
         self.message = message
-        self.displayed_message = f"[{self.status_code} Error] {self.message}"
+        self.displayed_message = f"\033[91m[{self.status_code} Error] {self.message}"
         super().__init__(self.displayed_message)
diff --git a/tests/test_qibo_client.py b/tests/test_qibo_client.py
@@ -126,7 +126,7 @@ def test_run_circuit_with_invalid_token(self, pass_version_check):
         with pytest.raises(exceptions.JobApiError) as err:
             self.obj.run_circuit(FAKE_CIRCUIT, FAKE_DEVICE, FAKE_NSHOTS)
 
-        expected_message = f"[404 Error] {message}"
+        expected_message = f"\033[91m[404 Error] {message}"
         assert str(err.value) == expected_message
 
     def test_run_circuit_with_job_post_error(self, pass_version_check):

diff --git a/tests/test_qibo_job.py b/tests/test_qibo_job.py
@@ -204,7 +204,7 @@ def test_refresh_with_invalid_pid(self):
         with pytest.raises(exceptions.JobApiError) as err:
             self.obj.refresh()
 
-        expected_message = f"[404 Error] {response_json['detail']}"
+        expected_message = f"\033[91m[404 Error] {response_json['detail']}"
         assert str(err.value) == expected_message
 
     @pytest.mark.parametrize(

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -52,7 +52,7 @@ def test_get_request_with_404_error():
     with pytest.raises(exceptions.JobApiError) as err:
         utils.QiboApiRequest.get(endpoint)
 
-    expected_message = f"[{status_code} Error] {message}"
+    expected_message = f"\033[91m[{status_code} Error] {message}"
     assert str(err.value) == expected_message
 
 
@@ -87,5 +87,5 @@ def test_post_request_with_404_error():
     with pytest.raises(exceptions.JobApiError) as err:
         utils.QiboApiRequest.post(endpoint)
 
-    expected_message = f"[{status_code} Error] {message}"
+    expected_message = f"\033[91m[{status_code} Error] {message}"
     assert str(err.value) == expected_message