mirror of
https://github.com/wassname/pandas-ta.git
synced 2026-06-29 16:30:38 +08:00
Kevin Johnson
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
from unittest import TestCase, skip
|
|
from pandas import DataFrame
|
|
|
|
from .config import sample_data
|
|
|
|
|
|
class TestStatisticsExtension(TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.data = sample_data
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
del cls.data
|
|
|
|
def setUp(self): pass
|
|
def tearDown(self): pass
|
|
|
|
|
|
def test_entropy_ext(self):
|
|
self.data.ta.entropy(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "ENTP_10")
|
|
|
|
def test_kurtosis_ext(self):
|
|
self.data.ta.kurtosis(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "KURT_30")
|
|
|
|
def test_mad_ext(self):
|
|
self.data.ta.mad(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "MAD_30")
|
|
|
|
def test_median_ext(self):
|
|
self.data.ta.median(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "MEDIAN_30")
|
|
|
|
def test_quantile_ext(self):
|
|
self.data.ta.quantile(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "QTL_30_0.5")
|
|
|
|
def test_skew_ext(self):
|
|
self.data.ta.skew(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "SKEW_30")
|
|
|
|
def test_stdev_ext(self):
|
|
self.data.ta.stdev(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "STDEV_30")
|
|
|
|
def test_tos_stdevall_ext(self):
|
|
self.data.ta.tos_stdevall(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(list(self.data.columns[-7:]), [
|
|
"TOS_STDEVALL_LR",
|
|
"TOS_STDEVALL_L_1", "TOS_STDEVALL_U_1",
|
|
"TOS_STDEVALL_L_2", "TOS_STDEVALL_U_2",
|
|
"TOS_STDEVALL_L_3", "TOS_STDEVALL_U_3"
|
|
])
|
|
|
|
def test_variance_ext(self):
|
|
self.data.ta.variance(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "VAR_30")
|