diff --git a/src/data_morph/data/stats.py b/src/data_morph/data/stats.py index 8c267ca6..3a35003c 100644 --- a/src/data_morph/data/stats.py +++ b/src/data_morph/data/stats.py @@ -12,7 +12,7 @@ ) -def get_values(data: pd.DataFrame) -> SummaryStatistics: +def get_summary_statistics(data: pd.DataFrame) -> SummaryStatistics: """ Calculate the summary statistics for the given set of points. diff --git a/src/data_morph/morpher.py b/src/data_morph/morpher.py index cf945c63..ee5dee9f 100644 --- a/src/data_morph/morpher.py +++ b/src/data_morph/morpher.py @@ -10,7 +10,7 @@ import numpy as np import tqdm -from .data.stats import get_values +from .data.stats import get_summary_statistics from .plotting.animation import ( ease_in_out_quadratic, ease_in_out_sine, @@ -264,7 +264,9 @@ def _is_close_enough(self, df1: pd.DataFrame, df2: pd.DataFrame) -> bool: np.abs( np.subtract( *( - np.floor(np.array(get_values(data)) * 10**self.decimals) + np.floor( + np.array(get_summary_statistics(data)) * 10**self.decimals + ) for data in [df1, df2] ) ) diff --git a/src/data_morph/plotting/static.py b/src/data_morph/plotting/static.py index 212ca745..20ba7d17 100644 --- a/src/data_morph/plotting/static.py +++ b/src/data_morph/plotting/static.py @@ -10,7 +10,7 @@ import numpy as np from matplotlib.ticker import EngFormatter -from ..data.stats import get_values +from ..data.stats import get_summary_statistics from .style import plot_with_custom_style if TYPE_CHECKING: @@ -64,7 +64,7 @@ def plot( ax.xaxis.set_major_formatter(tick_formatter) ax.yaxis.set_major_formatter(tick_formatter) - res = get_values(data) + res = get_summary_statistics(data) labels = ('X Mean', 'Y Mean', 'X SD', 'Y SD', 'Corr.') locs = np.linspace(0.8, 0.2, num=len(labels)) diff --git a/tests/data/test_stats.py b/tests/data/test_stats.py index 43fb6fe0..d10857c3 100644 --- a/tests/data/test_stats.py +++ b/tests/data/test_stats.py @@ -1,7 +1,7 @@ """Test the stats module.""" from data_morph.data.loader import DataLoader -from data_morph.data.stats import get_values +from data_morph.data.stats import get_summary_statistics def test_stats(): @@ -9,7 +9,7 @@ def test_stats(): data = DataLoader.load_dataset('dino').data - stats = get_values(data) + stats = get_summary_statistics(data) assert stats.x_mean == data.x.mean() assert stats.y_mean == data.y.mean()