File size: 822 Bytes
a2afe2f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
"""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)
|