Updates tests

mrjbq7 · mrjbq7 · commit a4ee637470d7 · 2023-07-13T06:55:34.000-07:00
diff --git a/tests/test_abstract.py b/tests/test_abstract.py
@@ -324,18 +324,6 @@ def test_call_without_arguments():
         abstract.Function('SMA')(10)
 
 
-def test_call_first_exception():
-    inputs = {'close': np.array([np.nan, np.nan, np.nan])}
-
-    with pytest.raises(Exception, match="inputs are all NaN"):
-        abstract.SMA(inputs, timeperiod=2)
-
-    inputs = {'close': np.array([1.0, 2.0, 3.0])}
-
-    output = abstract.SMA(inputs, timeperiod=2)
-    expected = np.array([np.nan, 1.5, 2.5])
-    assert_array_equal(expected, output)
-
 def test_threading():
     import pandas as pd
     TEST_LEN_SHORT = 999
diff --git a/tests/test_func.py b/tests/test_func.py
@@ -33,6 +33,13 @@ def test_input_lengths():
         func.BOP(a1, a1, a1, a2)
 
 
+def test_input_allnans():
+    a = np.arange(20, dtype=float)
+    a[:] = np.nan
+    r = func.RSI(a)
+    assert np.all(np.isnan(r))
+
+
 def test_input_nans():
     a1 = np.arange(10, dtype=float)
     a2 = np.arange(10, dtype=float)
