Skip to content

Commit

Permalink
revised according to Paramveers suggestion
Browse files Browse the repository at this point in the history
  • Loading branch information
merari95 committed Jan 16, 2025
1 parent 72c5700 commit efa5f69
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions tests/test_plotting.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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
Expand Down

0 comments on commit efa5f69

Please sign in to comment.