mirror of
https://github.com/wassname/pandas-ta.git
synced 2026-06-28 16:20:17 +08:00
Kevin Johnson
113 lines
3.7 KiB
Python
113 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from unittest import TestCase, skip
|
|
from pandas import DataFrame
|
|
|
|
from .config import sample_data
|
|
|
|
|
|
class TestVolumeExtension(TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.data = sample_data
|
|
cls.open = cls.data["open"]
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
del cls.data
|
|
del cls.open
|
|
|
|
def setUp(self): pass
|
|
def tearDown(self): pass
|
|
|
|
|
|
def test_ad_ext(self):
|
|
self.data.ta.ad(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "AD")
|
|
|
|
def test_ad_open_ext(self):
|
|
self.data.ta.ad(open_=self.open, append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "ADo")
|
|
|
|
def test_adosc_ext(self):
|
|
self.data.ta.adosc(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "ADOSC_3_10")
|
|
|
|
def test_aobv_ext(self):
|
|
self.data.ta.aobv(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(
|
|
list(self.data.columns[-7:]),
|
|
["OBV", "OBV_min_2", "OBV_max_2", "OBVe_4", "OBVe_12", "AOBV_LR_2", "AOBV_SR_2"],
|
|
)
|
|
# Remove "OBV" so it does not interfere with test_obv_ext()
|
|
self.data.drop("OBV", axis=1, inplace=True)
|
|
|
|
def test_cmf_ext(self):
|
|
self.data.ta.cmf(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "CMF_20")
|
|
|
|
def test_efi_ext(self):
|
|
self.data.ta.efi(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "EFI_13")
|
|
|
|
def test_eom_ext(self):
|
|
self.data.ta.eom(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "EOM_14_100000000")
|
|
|
|
def test_kvo_ext(self):
|
|
self.data.ta.kvo(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(list(self.data.columns[-2:]), ["KVO_34_55_13", "KVOs_34_55_13"])
|
|
|
|
def test_mfi_ext(self):
|
|
self.data.ta.mfi(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "MFI_14")
|
|
|
|
def test_nvi_ext(self):
|
|
self.data.ta.nvi(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "NVI_1")
|
|
|
|
def test_obv_ext(self):
|
|
self.data.ta.obv(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "OBV")
|
|
|
|
def test_pvi_ext(self):
|
|
self.data.ta.pvi(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "PVI_1")
|
|
|
|
def test_pvol_ext(self):
|
|
self.data.ta.pvol(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "PVOL")
|
|
|
|
def test_pvr_ext(self):
|
|
self.data.ta.pvr(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "PVR")
|
|
|
|
def test_pvt_ext(self):
|
|
self.data.ta.pvt(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "PVT")
|
|
|
|
@skip("\nVP does not return a Time Series")
|
|
def test_vp_ext(self):
|
|
result = self.data.ta.vp()
|
|
self.assertIsInstance(result, DataFrame)
|
|
self.assertEqual(result.name, "VP_10")
|
|
|
|
def test_wb_tsv_ext(self):
|
|
self.data.ta.wb_tsv(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(list(self.data.columns[-3:]), ["TSV_18_10", "TSVs_18_10", "TSVr_18_10"])
|