mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-05-11 10:13:38 +08:00
a255fa52a0
* PyDocstyle Compliant batch 1 * Batch 2 * Batch 3 * Batch 4 * Batch 5 * Batch 6 * Batch 7 * Linting * MyPy * mypy things * black * more mypy * black * fix that * black again * undo that * Fix error message * pylint --------- Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com> Co-authored-by: Danglewood <85772166+deeleeramone@users.noreply.github.com>
33 lines
822 B
Python
33 lines
822 B
Python
"""Tests for the statistics module."""
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
from openbb_quantitative.statistics import kurtosis_, mean_, skew_, std_dev_, var_
|
|
|
|
test_data = pd.Series([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
|
|
|
|
|
|
def test_kurtosis():
|
|
"""Test the kurtosis function."""
|
|
assert kurtosis_(test_data) == pytest.approx(-1.224, abs=1e-3)
|
|
|
|
|
|
def test_skew():
|
|
"""Test the skewness function."""
|
|
assert skew_(test_data) == pytest.approx(0.0, abs=1e-3)
|
|
|
|
|
|
def test_std_dev():
|
|
"""Test the standard deviation function."""
|
|
assert std_dev_(test_data) == pytest.approx(2.872, abs=1e-3)
|
|
|
|
|
|
def test_mean():
|
|
"""Test the mean function."""
|
|
assert mean_(test_data) == pytest.approx(5.5, abs=1e-3)
|
|
|
|
|
|
def test_var():
|
|
"""Test the variance function."""
|
|
assert var_(test_data) == pytest.approx(8.25, abs=1e-3)
|