mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 04:52:07 +00:00
cd7d602b8f
* refactor quantitative for v5 * add factors endpoint and add a few charts --------- Co-authored-by: deeleeramone <>
40 lines
869 B
Python
40 lines
869 B
Python
"""Tests for ``openbb_quantitative._stats_helpers``."""
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
|
|
from openbb_quantitative._stats_helpers 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)
|