print debug

robertmartin8 · Dec 4, 2023 · ea7e16c · ea7e16c
1 parent fb48396
commit ea7e16c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/tests/test_efficient_frontier.py b/tests/test_efficient_frontier.py
@@ -1022,7 +1022,12 @@ def test_efficient_risk_market_neutral_warning():
     ef = setup_efficient_frontier()
     with pytest.warns(RuntimeWarning) as w:
         ef.efficient_risk(0.19, market_neutral=True)
-        print("test_efficient_risk_market_neutral_warning", len(w), type(w), str(w))
+        print(
+            "test_efficient_risk_market_neutral_warning",
+            len(w),
+            type(w),
+            [v.message for v in w],
+        )
         assert len(w) == 1
         assert (
             str(w[0].message)
@@ -1187,7 +1192,12 @@ def test_efficient_return_market_neutral_warning():
     ef = setup_efficient_frontier()
     with pytest.warns(RuntimeWarning) as w:
         ef.efficient_return(0.25, market_neutral=True)
-        print("test_efficient_return_market_neutral_warning", len(w), type(w), str(w))
+        print(
+            "test_efficient_return_market_neutral_warning",
+            len(w),
+            type(w),
+            [v.message for v in w],
+        )
         assert len(w) == 1
         assert (
             str(w[0].message)

diff --git a/tests/test_plotting.py b/tests/test_plotting.py
@@ -72,7 +72,7 @@ def test_dendrogram_plot():
     hrp = HRPOpt(returns)
     with pytest.warns(RuntimeWarning) as w:
         ax = plotting.plot_dendrogram(hrp, show_tickers=False, showfig=False)
-        print("test_dendrogram_plot", len(w), type(w), str(w))
+        print("test_dendrogram_plot", len(w), type(w), [v.message for v in w])
         assert len(w) == 1
         assert (
             str(w[0].message)