also check sliding_window for pandas series

MaxBenChrist · MaxBenChrist · commit ee21da274680 · 2017-08-18T20:19:28.000+02:00
diff --git a/tests/feature_extraction/test_motifs.py b/tests/feature_extraction/test_motifs.py
@@ -9,7 +9,7 @@
 import numpy as np
 import numpy.testing as npt
 from tsfresh.feature_extraction import motifs
-
+import pandas as pd
 
 class TestMotifSubelements(unittest.TestCase):
     def setUp(self):
@@ -54,6 +54,10 @@ def test_sliding_window(self):
         expected_result = np.array([[0, 1], [1, 2], [2, 3], [3, 4]])
         npt.assert_array_equal(expected_result, motifs._sliding_window(data, pattern_length=2))
 
+        data = pd.Series(np.arange(5))
+        expected_result = np.array([[0, 1], [1, 2], [2, 3], [3, 4]])
+        npt.assert_array_equal(expected_result, motifs._sliding_window(data, pattern_length=2))
+
     def test_match_scores(self):
         answer = motifs._match_scores(self.s1, self.s1[5:10])
         self.assertEqual(answer[5], 0.0)
