mirror of
https://github.com/wassname/pandas-ta.git
synced 2026-07-04 17:20:12 +08:00
Kevin Johnson
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
from unittest import TestCase, skip
|
|
from pandas import DataFrame
|
|
|
|
from .config import sample_data
|
|
|
|
|
|
class TestCandleExtension(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_cdl_doji_ext(self):
|
|
self.data.ta.cdl_pattern("doji", append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "CDL_DOJI_10_0.1")
|
|
|
|
def test_cdl_inside_ext(self):
|
|
self.data.ta.cdl_pattern("inside", append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(self.data.columns[-1], "CDL_INSIDE")
|
|
|
|
def test_cdl_z_ext(self):
|
|
self.data.ta.cdl_z(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(list(self.data.columns[-4:]), ["open_Z_30_1", "high_Z_30_1", "low_Z_30_1", "close_Z_30_1"])
|
|
|
|
def test_ha_ext(self):
|
|
self.data.ta.ha(append=True)
|
|
self.assertIsInstance(self.data, DataFrame)
|
|
self.assertEqual(list(self.data.columns[-4:]), ["HA_open", "HA_high", "HA_low", "HA_close"])
|