From efa5f691516e01228e8725cddb1f817ca04be14f Mon Sep 17 00:00:00 2001 From: Merari Santana Date: Wed, 15 Jan 2025 19:07:58 -0800 Subject: [PATCH] revised according to Paramveers suggestion --- tests/test_plotting.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/test_plotting.py b/tests/test_plotting.py index 825379b..ccf1d40 100644 --- a/tests/test_plotting.py +++ b/tests/test_plotting.py @@ -2,10 +2,9 @@ import pandas as pd import numpy as np from altair import Chart -import os -import sys -sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'src', 'linreg_ally')) +from linreg_ally.plotting import qq_and_residuals_plot from plotting import qq_and_residuals_plot +from altair import HConcatChart # Unit Tests @@ -19,7 +18,6 @@ def test_qq_and_residuals_plot_valid_inputs(): # Test with concatenate=True result = qq_and_residuals_plot(y_actual, y_predicted, concatenate=True) - from altair import HConcatChart # Import HConcatChart assert isinstance(result, (Chart, HConcatChart)), "Expected an Altair Chart or HConcatChart when concatenate=True" # Test with concatenate=False