docker-python/tests/test_tsfresh.py at master · feibyte/docker-python · GitHub

18 lines (13 loc) · 496 Bytes

import unittest

import pandas as pd

import numpy as np

from tsfresh import extract_features

class TestTsFresh(unittest.TestCase):

def test_extract_feature(self):

ts = pd.DataFrame({

'id': np.array(['a', 'a', 'a', 'b', 'b', 'b']),

'time': np.array([0,1,2,0,1,2]),

'x': np.array([3,4,5,7,8,10])

})

extracted_features = extract_features(ts, column_id='id', column_sort='time')

self.assertEqual(2, len(extracted_features))

Expand file tree