STY: Replaced hardcoded values with variables.

2026-07-13 17:42:42 +08:00 · 2012-12-06 15:26:49 -05:00
parent c7383f6275
commit a14d702a0d
1 changed files with 3 additions and 1 deletions
@@ -341,7 +341,9 @@ class TestBatchTransform(TestCase):
        # test overloaded class
        for test_history in [algo.history_return_price_class,
                             algo.history_return_price_decorator]:
-            for i in range(3, 6):
+            # starting at window length, the window should contain
+            # consecutive (of window length) numbers up till the end.
+            for i in range(algo.window_length, len(test_history)):
                np.testing.assert_array_equal(
                    range(i - algo.window_length + 1, i + 1),
                    test_history[i].values.flatten()