Remove pandas helper function, move to superstore

streamlet-dev · Feb 10, 2024 · 1ec60d2 · 1ec60d2
1 parent 6d638f9
commit 1ec60d2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 16 deletions.
diff --git a/setup.py b/setup.py
@@ -61,6 +61,7 @@ def get_version(file, name="__version__"):
     "pybind11>=2.4.0",
     "pytest>=4.3.0",
     "pytest-cov>=2.6.1",
+    "superstore",
     "Sphinx>=1.8.4",
     "sphinx-markdown-builder>=0.5.2",
 ] + requires

diff --git a/tributary/tests/lazy/calculations/test_finance_lazy.py b/tributary/tests/lazy/calculations/test_finance_lazy.py
@@ -1,13 +1,11 @@
 import pandas as pd
+import superstore
 import tributary.lazy as tl
 
 
 class TestFinance:
     def test_rsi(self):
-        try:
-            df = pd.DataFrame(pd.util.testing.getTimeSeriesData(20))
-        except AttributeError:
-            df = pd.DataFrame(pd._testing.getTimeSeriesData(20))
+        df = pd.DataFrame(superstore.getTimeSeriesData(20))
         adjust = False
         period = 14
         delta = df["A"].diff().shift(-1)
@@ -34,10 +32,7 @@ def test_rsi(self):
         assert n_rsi().tolist() == rsi.tolist()
 
     def test_macd(self):
-        try:
-            df = pd.DataFrame(pd.util.testing.getTimeSeriesData(20))
-        except AttributeError:
-            df = pd.DataFrame(pd._testing.getTimeSeriesData(20))
+        df = pd.DataFrame(superstore.getTimeSeriesData(20))
 
         period_fast = 12
         period_slow = 26

diff --git a/tributary/tests/streaming/calculations/test_finance_streaming.py b/tributary/tests/streaming/calculations/test_finance_streaming.py
@@ -1,13 +1,11 @@
 import pandas as pd
+import superstore
 import tributary.streaming as ts
 
 
 class TestFinance:
     def test_rsi(self):
-        try:
-            vals = pd.DataFrame(pd.util.testing.getTimeSeriesData(20))
-        except AttributeError:
-            vals = pd.DataFrame(pd._testing.getTimeSeriesData(20))
+        vals = pd.DataFrame(superstore.getTimeSeriesData(20))
         adjust = False
         period = 14
         delta = vals["A"].diff().shift(-1)
@@ -28,10 +26,7 @@ def test_rsi(self):
             assert abs(x - y) < 0.001
 
     def test_macd(self):
-        try:
-            vals = pd.DataFrame(pd.util.testing.getTimeSeriesData(20))
-        except AttributeError:
-            vals = pd.DataFrame(pd._testing.getTimeSeriesData(20))
+        vals = pd.DataFrame(superstore.getTimeSeriesData(20))
         period_fast = 12
         period_slow = 26
         signal = 9