QuanTAlib/python/tests/__pycache__/test_compat.cpython-313-pytest-9.0.2.pyc

ó
±7£i+ã
óx
•SrS
SK
Jr S
SKrS
SKJs Jr S
SK	r
S
SKr\
RRS5
rSr\R!\5
R#5S-r\R&"SS9
S	5r\R&"SS9
S
5r"SS5r"S
S5r"SS5r"SS5r"SS5rg)uØtest_compat.py — pandas-ta compatibility tests.

Verifies that:
1. ALIASES map resolves to real functions
2. pd.Series input → pd.Series output with correct name
3. pd.DataFrame input → works for single-column
é)
ÚannotationsNécé2gY@Úmodule)
Úscopec
ó|•S
SKnU$![[4a#n
[R"SU
35
 Sn
A
gSn
A
f
f=f
)Nrz$quantalib native lib not available: )Ú	quantalibÚOSErrorÚImportErrorÚpytestÚskip)Ú_qtlÚ
es  Ú/c:\github\quantalib\python\tests\test_compat.pyÚqtlrs>€ð@
Û ØˆøÜ”[Ð!ó
@
Ü�ŠÐ:¸1¸#Ð>×?Ñ?ûð
@
ús‚ˆ;˜6¶;c
ó\•S
SKnU$![a [R"S5
 gf=f
)Nrzpandas not installed)Úpandasrrr
)
Ú_pds
 rÚpdrs+€ð,Û؈
øÜó
,Ü�ŠÐ*Ö+ð
,ús‚ˆ +ª
+c
ó6•\r
SrS
rSrSSjrSSjrSSjrSrg)	ÚTestAliasesé$z5Verify ALIASES map entries resolve to real functions.c
ó•S
SKJ
n UR5Híup4[U
RUS5nSoeUL
ow(dÃ[
R"SU4
SXV45S[R"5;d[
R"U5
(a[
R"U5
O
S[
R"U5
S.-n[
R"SUSUS	35
S
-SU0
-n	[[
R"U	5
5
e
S=
pvMï g)Nr)
ÚALIASES)
úis not)
z%(py0)s is not %(py3)sÚfn)Úpy0Úpy3zAlias 'u' → 'z' not foundz
>assert %(py5)sÚpy5)Úquantalib._compatrÚitemsÚgetattrÚ
indicatorsÚ
@pytest_arÚ_call_reprcompareÚ@py_builtinsÚlocalsÚ_should_repr_global_nameÚ	_safereprÚ_format_assertmsgÚAssertionErrorÚ_format_explanation)
ÚselfrrÚaliasÚtargetrÚ@py_assert2Ú@py_assert1Ú@py_format4Ú@py_format6s
          rÚtest_all_aliases_resolveÚ$TestAliases.test_all_aliases_resolve's €Ý-Ø$Ÿ]™]ž_‰MˆEܘŸ™¨°Ó6ˆBØ!ÐN˜T�>×N×N×NÒN�2×N×N×N×N×NÕN�2×N×N×NÑN�2×N×NÕN˜T×N×N×NÐN W¨U¨G°7¸6¸(À+Ð#N×N×N×N×N×N×NÐNò-óc
óö
•S
SKJ
n U"S5
n[U5
oD(dØSS[R"5;d[
R"[5
(a[
R"[5
O
SS[R"5;d[
R"U5
(a[
R"U5
O
S[
R"U5
S.-n[[
R"U5
5
e
Sng)Nr©
Ú
get_compatÚmidpricez,assert %(py3)s
{%(py3)s = %(py0)s(%(py1)s)
}Úcallabler)rÚpy1r)
r r9r;r&r'r$r(r)r+r,)r-rr9rr0r2s      rÚ test_get_compat_returns_callableÚ,TestAliases.test_get_compat_returns_callable-s|€Ý0Ù
˜
Ó
#ˆÜ˜‹|Ð�|×××××ÑŒx×××ÑŒx×××××Ö˜×××ј××Õˆ|×××××Òr6c
óä
•S
SKJ
n So2"U5
nSoTULof(dÏ[R"SU4
SXE45S[R
"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U5
[R"U5
S.-nSS	U0
-n[[R"U5
5
e
S=
n=
n=
peg)
Nrr8Únonexistent_indicator)
Úis)
z0%(py4)s
{%(py4)s = %(py0)s(%(py2)s)
} is %(py7)sr9)rÚpy2Úpy4Úpy7zassert %(py9)sÚpy9)
r r9r$r%r&r'r(r)r+r,)	r-rr9r1Ú@py_assert3Ú@py_assert6Ú@py_assert5Ú@py_format8Ú@py_format10s	         rÚ$test_get_compat_unknown_returns_noneÚ0TestAliases.test_get_compat_unknown_returns_none2s‰€Ý0Ø1Ð:ˆzÐ1Ó2Ð:°dÐ:°dÐ:×:×:×:Ò:Ð2×:×:×:×:×:Õ:ˆz×:×:×:Ñ:ˆz×:×:Õ:Ð1×:×:Ó:Ð2×:×:Ó:°d×:×:×:×:×:×:Ö:r6©N©ÚreturnÚNone)	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__r4r=rKÚ__static_attributes__rMr6rrr$s†Ù?ôO
ô÷
;r6rc
ó6•\r
SrS
rSrSSjrSSjrSSjrSrg)	ÚTestPandasSeriesIOé7u,Verify pd.Series input → pd.Series output.c
óD
•UR
S
[SS9nUR[USS9nU
R	USS9nURn[XV5ow(G
d8SS	[R"5;d[R"[
5
(a[R"[
5
O
S	S
[R"5;d[R"U5
(a[R"U5
O
S
S[R"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U5
S.-n[[R"U5
5
e
S=
pgURn	S
o©U
:Hof(dº[R"SU4
SXš45S
[R"5;d[R"U5
(a[R"U5
O
S
[R"U	5
[R"U
5
S.-nSSU0
-n[[R"U5
5
e
S=
n	=
pj[U5
oÝ[:Hoª(G
dP[R"SU
4
SU
[45S[R"5;d[R"[5
(a[R"[5
O
SS
[R"5;d[R"U5
(a[R"U5
O
S
[R"U
5
S[R"5;d[R"[5
(a[R"[5
O
SS.-nSSU0
-n[[R"U5
5
e
S=
pÚUR o™U:HofR"oî"5oÿ(G
d[R"SU4
SX“45S
[R"5;d[R"U5
(a[R"U5
O
S
[R"U	5
S[R"5;d[R"U5
(a[R"U5
O
SS.-nSU[R"U5
[R"U5
S.-n[[R"U5
5
e
S=
n	=
n=
pïg)Nz
2020-01-01Ú
D)ÚperiodsÚfreqÚClose)ÚindexÚnameé
©
ÚlengthúQassert %(py6)s
{%(py6)s = %(py0)s(%(py1)s, %(py4)s
{%(py4)s = %(py2)s.Series
})
}Ú
isinstanceÚresultr©rr<rBrCÚpy6ÚSMA_10©
ú==©
z,%(py2)s
{%(py2)s = %(py0)s.name
} == %(py5)s©rrBrúassert %(py7)srD©
z0%(py3)s
{%(py3)s = %(py0)s(%(py1)s)
} == %(py5)sÚlenÚ
N©rr<rr)
z-%(py2)s
{%(py2)s = %(py0)s.index
} == %(py4)sÚidx)rrBrCz@assert %(py10)s
{%(py10)s = %(py8)s
{%(py8)s = %(py6)s.all
}()
})rhÚpy8Úpy10)Ú
date_rangerqÚSeriesÚCLOSEÚsmarer&r'r$r(r)r+r,r`r%rpr_Úall)r-rrrsÚ
srfrFrHÚ@py_format7r1Ú@py_assert4r3rIr0Ú@py_assert7Ú@py_assert9Ú@py_format5Ú@py_format11s                  rÚtest_sma_series_outputÚ)TestPandasSeriesIO.test_sma_series_output:s¤€Ø�m‰m˜L´!¸#ˆmÐ>ˆØ�I‰I”e 3¨WˆIÐ5ˆ
Ø—‘˜ 2�Ð&ˆØ"$§)¡)Ð,Œz˜&Ó,Ð,Ö,×,×,×,×,×,Ñ,Œz×,×,×,Ñ,Œz×,×,×,×,×,Ö,˜&×,×,×,Ñ,˜&×,×,×,×,×,Ö, "×,×,×,Ñ, "×,×,Õ, )×,×,Ó,Ð,×,×,×,×,×,Ò,Ø�{‰{Ð&˜hÐ&˜hÑ&×&×&×&Ò&ˆ{×&×&×&×&×&Õ&ˆv×&×&×&Ñ&ˆv×&×&Õ&ˆ{×&×&Ó&˜h×&×&×&×&×&×&Ó&Ü�6‹{ÐœaÑ×××Óˆ{œa×××××ÕŒs×××ÑŒs×××××Ö�6×××Ñ�6××Õˆ{×××××Ôœa×××Ñœa××××××ÓØ—‘Ð* Ñ#Ð*×(Ñ(Ð*Ð(Ó*Ð*Ö*×*×*Ó*�×*×*×*×*×*Õ*�×*×*×*Ñ*�×*×*Õ*�×*×*×*×*×*Ô* ×*×*×*Ñ* ×*×*×*Ô*Ð(×*×*Ó*Ð*×*×*×*×*×*×*r6c
ób•UR
[5
nU
RUS
S9nURn[XE5of(G
d8SS[R
"5;d[R"[5
(a[R"[5
O
SS[R
"5;d[R"U5
(a[R"U5
O
SS[R
"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U5
S.-n[[R"U5
5
e
S=
pVURnSo˜U	:HoU(dº[R"S	U4
S
X‰45S[R
"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U	5
S.-n
SS
U
0
-n[[R"U5
5
e
S=
n=
pYSn[XL5o™(díSS[R
"5;d[R"[5
(a[R"[5
O
SS[R
"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U	5
S.-n
[[R"U
5
5
e
S=
pÉURnSo˜U	:HoU(dº[R"S	U4
SX‰45S[R
"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U	5
S.-n
SS
U
0
-n[[R"U5
5
e
S=
n=
pYg)NérbrdrerfrrgÚEMA_14rjrlrmrnrDÚcategoryz5assert %(py5)s
{%(py5)s = %(py0)s(%(py1)s, %(py3)s)
}ÚhasattrrrÚtrend)
z0%(py2)s
{%(py2)s = %(py0)s.category
} == %(py5)s)rwrxÚemarer&r'r$r(r)r+r,r`r%rˆr‡)
r-rrr{rfrFrHr|r1r}r3rIr0s
             rÚtest_ema_series_categoryÚ+TestPandasSeriesIO.test_ema_series_categoryCs&€Ø�I‰I”eÓˆ
Ø—‘˜ 2�Ð&ˆØ"$§)¡)Ð,Œz˜&Ó,Ð,Ö,×,×,×,×,×,Ñ,Œz×,×,×,Ñ,Œz×,×,×,×,×,Ö,˜&×,×,×,Ñ,˜&×,×,×,×,×,Ö, "×,×,×,Ñ, "×,×,Õ, )×,×,Ó,Ð,×,×,×,×,×,Ò,Ø�{‰{Ð&˜hÐ&˜hÑ&×&×&×&Ò&ˆ{×&×&×&×&×&Õ&ˆv×&×&×&Ñ&ˆv×&×&Õ&ˆ{×&×&Ó&˜h×&×&×&×&×&×&Ó&Ø)Ð*Œw�vÓ*Ð*Õ*×*×*×*×*×*Ñ*Œw×*×*×*Ñ*Œw×*×*×*×*×*Ö*�v×*×*×*Ñ*�v×*×*Õ*˜z×*×*Ó*Ð*×*×*×*×*×*Ò*Ø�‰Ð) 'Ð) 'Ñ)×)×)×)Ò)ˆ×)×)×)×)×)Õ)ˆv×)×)×)Ñ)ˆv×)×)Õ)ˆ×)×)Ó) '×)×)×)×)×)×)Ô)r6c
ó®•UR
[5
nU
RUS
S9nURn[XE5of(G
d8SS[R
"5;d[R"[5
(a[R"[5
O
SS[R
"5;d[R"U5
(a[R"U5
O
SS[R
"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U5
S.-n[[R"U5
5
e
S=
pVURnSo˜U	:HoU(dº[R"S	U4
S
X‰45S[R
"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U	5
S.-n
SS
U
0
-n[[R"U5
5
e
S=
n=
pYg)Nr…rbrdrerfrrgÚRSI_14rjrlrmrnrD)
rwrxÚrsirer&r'r$r(r)r+r,r`r%)r-rrr{rfrFrHr|r1r}r3rIs            rÚtest_rsi_seriesÚ"TestPandasSeriesIO.test_rsi_seriesKs7
€Ø�I‰I”eÓˆ
Ø—‘˜ 2�Ð&ˆØ"$§)¡)Ð,Œz˜&Ó,Ð,Ö,×,×,×,×,×,Ñ,Œz×,×,×,Ñ,Œz×,×,×,×,×,Ö,˜&×,×,×,Ñ,˜&×,×,×,×,×,Ö, "×,×,×,Ñ, "×,×,Õ, )×,×,Ó,Ð,×,×,×,×,×,Ò,Ø�{‰{Ð&˜hÐ&˜hÑ&×&×&×&Ò&ˆ{×&×&×&×&×&Õ&ˆv×&×&×&Ñ&ˆv×&×&Õ&ˆ{×&×&Ó&˜h×&×&×&×&×&×&Ô&r6rMNrN)	rQrRrSrTrUr‚r‹r�rVrMr6rrXrX7s†Ù6ô+ô*÷'r6rXc
ó"•\r
SrS
rSrSSjrSrg)ÚTestPandasDataFrameIOéRz,Verify pd.DataFrame input uses first column.c
ó•UR
[[R"[5
S
.5
nU
RUSS9nURn[XE5of(G
d8SS[R"5;d[R"[5
(a[R"[5
O
SS[R"5;d[R"U5
(a[R"U5
O
SS[R"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U5
S.-n[[R"U5
5
e
S=
pV[U5
oˆ[:Ho™(G
dP[R "S	U	4
S
U[45S[R"5;d[R"[5
(a[R"[5
O
SS[R"5;d[R"U5
(a[R"U5
O
S[R"U5
S[R"5;d[R"[5
(a[R"[5
O
SS
.-n
SSU
0
-n[[R"U5
5
e
S=
p‰g)N)r^ÚVolumerarbrdrerfrrgrjrorprqrrrnrD)Ú	DataFramerxÚnpÚonesrqryrwrer&r'r$r(r)r+r,rpr%)r-rrÚdfrfrFrHr|r0r}r3rIs            rÚtest_sma_dataframe_inputÚ.TestPandasDataFrameIO.test_sma_dataframe_inputUsz
ۯ
�\‰\¤E´R·W²W¼Q³ZÑ@Ó
AˆØ—‘˜ B�Ð'ˆØ"$§)¡)Ð,Œz˜&Ó,Ð,Ö,×,×,×,×,×,Ñ,Œz×,×,×,Ñ,Œz×,×,×,×,×,Ö,˜&×,×,×,Ñ,˜&×,×,×,×,×,Ö, "×,×,×,Ñ, "×,×,Õ, )×,×,Ó,Ð,×,×,×,×,×,Ò,Ü�6‹{ÐœaÑ×××Óˆ{œa×××××ÕŒs×××ÑŒs×××××Ö�6×××Ñ�6××Õˆ{×××××Ôœa×××Ñœa××××××Ôr6rMNrN)rQrRrSrTrUr›rVrMr6rr“r“Rs
†Ù6÷ r6r“c
ó"•\r
SrS
rSrSSjrSrg)ÚTestMultiOutputPandasé\z8Verify multi-output returns DataFrame when given Series.c
ó	•UR
[5
nU
RUS
SS9nURn[	XE5of(G
d8SS[
R"5;d[R"[5
(a[R"[5
O
SS[
R"5;d[R"U5
(a[R"U5
O
SS[
R"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U5
S.-n[[R"U5
5
e
S=
pVURn[S	4o˜U	:HoU(dº[R"S
U4
SX‰45S[
R"5;d[R"U5
(a[R"U5
O
S[R"U5
[R"U	5
S.-n
S
SU
0
-n[[R"U5
5
e
S=
n=
pY[UR 5
nSoÝU;oî(d¥[R"SU4
SXÜ45[R"U
5
S[
R"5;d[R"U5
(a[R"U5
O
SS.-nSSU0
-n
[[R"U
5
5
e
S=
pÞSoÝU;oî(d¥[R"SU4
SXÜ45[R"U
5
S[
R"5;d[R"U5
(a[R"U5
O
SS.-nSSU0
-n
[[R"U
5
5
e
S=
pÞSoÝU;oî(d¥[R"SU4
SXÜ45[R"U
5
S[
R"5;d[R"U5
(a[R"U5
O
SS.-nSSU0
-n
[[R"U
5
5
e
S=
pÞg)Nég@)rcÚstdzTassert %(py6)s
{%(py6)s = %(py0)s(%(py1)s, %(py4)s
{%(py4)s = %(py2)s.DataFrame
})
}rerfrrgérj)
z-%(py2)s
{%(py2)s = %(py0)s.shape
} == %(py5)srmrnrDz
BBU_20_2.0)
Úin)
z%(py1)s in %(py3)sÚcols)r<rzassert %(py5)srz
BBM_20_2.0z