diff --git a/openbb_terminal/stocks/comparison_analysis/ca_controller.py b/openbb_terminal/stocks/comparison_analysis/ca_controller.py index c5117b83416..5cec59be404 100644 --- a/openbb_terminal/stocks/comparison_analysis/ca_controller.py +++ b/openbb_terminal/stocks/comparison_analysis/ca_controller.py @@ -598,6 +598,7 @@ class ComparisonAnalysisController(BaseController): similar_tickers=self.similar, start=ns_parser.start.strftime("%Y-%m-%d"), candle_type=ns_parser.type_candle, + export=ns_parser.export, ) else: console.print("Please make sure there are similar tickers selected. \n") diff --git a/openbb_terminal/stocks/comparison_analysis/yahoo_finance_view.py b/openbb_terminal/stocks/comparison_analysis/yahoo_finance_view.py index 9f9d8c72a66..f8ce23f2de3 100644 --- a/openbb_terminal/stocks/comparison_analysis/yahoo_finance_view.py +++ b/openbb_terminal/stocks/comparison_analysis/yahoo_finance_view.py @@ -168,6 +168,7 @@ def display_correlation( start: str = (datetime.now() - timedelta(days=366)).strftime("%Y-%m-%d"), candle_type: str = "a", external_axes: Optional[List[plt.Axes]] = None, + export: str = "", ): """ Correlation heatmap based on historical price comparison @@ -183,6 +184,8 @@ def display_correlation( OHLCA column to use, by default "a" for Adjusted Close external_axes : Optional[List[plt.Axes]], optional External axes (1 axis is expected in the list), by default None + export : str, optional + Format to export correlation prices, by default "" """ df_similar = yahoo_finance_model.get_historical(similar_tickers, start, candle_type) @@ -227,4 +230,5 @@ def display_correlation( if not external_axes: theme.visualize_output() + export_data(export, os.path.dirname(os.path.abspath(__file__)), "hcorr", df_similar) console.print("") diff --git a/tests/openbb_terminal/stocks/comparison_analysis/test_ca_controller.py b/tests/openbb_terminal/stocks/comparison_analysis/test_ca_controller.py index d33c0ab1793..c05ea48fada 100644 --- a/tests/openbb_terminal/stocks/comparison_analysis/test_ca_controller.py +++ b/tests/openbb_terminal/stocks/comparison_analysis/test_ca_controller.py @@ -256,6 +256,7 @@ def test_call_func_expect_queue(expected_queue, queue, func): similar_tickers=["MOCK_SIMILAR_1", "MOCK_SIMILAR_2"], start="2020-12-01", candle_type="h", + export="", ), ), (