Fix test for abstract w/ polars.

mrjbq7 · mrjbq7 · commit e38ade10be58 · 2021-10-24T09:23:29.000-07:00
diff --git a/talib/test_polars.py b/talib/test_polars.py
@@ -3,6 +3,7 @@
 from nose.tools import assert_equals, assert_is_instance, assert_true
 
 import talib
+from talib import abstract
 from talib.test_data import series, assert_np_arrays_equal
 
 def test_MOM():
@@ -51,6 +52,6 @@ def test_TEVA():
             "volume": np.random.uniform(low=0.0, high=100.0, size=size).astype("float32")
         }
     )
-    tema = ta.TEMA(df, timeperiod=9)
+    tema = abstract.TEMA(df, timeperiod=9)
     assert_is_instance(tema, pl.Series)
     assert_equal(len(tema), 50)

Original file line number	Diff line number	Diff line change
`@@ -3,6 +3,7 @@`
`3`	`3`	`from nose.tools import assert_equals, assert_is_instance, assert_true`
`4`	`4`
`5`	`5`	`import talib`
	`6`	`+from talib import abstract`
`6`	`7`	`from talib.test_data import series, assert_np_arrays_equal`
`7`	`8`
`8`	`9`	`def test_MOM():`
`@@ -51,6 +52,6 @@ def test_TEVA():`
`51`	`52`	`"volume": np.random.uniform(low=0.0, high=100.0, size=size).astype("float32")`
`52`	`53`	`}`
`53`	`54`	`)`
`54`		`- tema = ta.TEMA(df, timeperiod=9)`
	`55`	`+ tema = abstract.TEMA(df, timeperiod=9)`
`55`	`56`	`assert_is_instance(tema, pl.Series)`
`56`	`57`	`assert_equal(len(tema), 50)`