You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
644 B
23 lines
644 B
6 years ago
|
import unittest
|
||
|
|
from detectors.pattern_detector import PatternDetector
|
||
|
|
|
||
|
|
def rlist(start, stop):
|
||
|
|
return [x for x in range(start, stop + 1)]
|
||
|
|
class TestUtils(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_chunks_generator(self):
|
||
|
|
window_size = 1
|
||
|
|
|
||
|
|
cases = [
|
||
|
|
([], [[]]),
|
||
|
|
(rlist(0, 300), [rlist(0,99),rlist(99,198),rlist(198,297),rlist(297,300)])
|
||
|
|
]
|
||
|
|
|
||
|
|
for data, expected_chunks in cases:
|
||
|
|
chunks = tuple(PatternDetector._PatternDetector__get_data_chunks(None, data, window_size))
|
||
|
|
self.assertSequenceEqual(chunks, expected_chunks)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|