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.
44 lines
1.3 KiB
44 lines
1.3 KiB
5 years ago
|
import unittest
|
||
|
|
from utils import get_intersected_chunks, get_chunks
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
|
||
|
|
class TestUtils(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_chunks_generator(self):
|
||
|
|
intersection = 2
|
||
|
|
chunk_size = 4
|
||
|
|
|
||
|
|
cases = [
|
||
|
|
(list(range(8)), [[0,1,2,3], [2,3,4,5], [4,5,6,7]]),
|
||
|
|
([], [[]]),
|
||
|
|
(list(range(1)), [[0]]),
|
||
|
|
(list(range(4)), [[0,1,2,3]]),
|
||
|
|
(list(range(9)), [[0,1,2,3], [2,3,4,5], [4,5,6,7], [6,7,8]])
|
||
|
|
]
|
||
|
|
|
||
|
|
for tested, expected in cases:
|
||
|
|
tested_chunks = get_intersected_chunks(tested, intersection, chunk_size)
|
||
|
|
self.assertSequenceEqual(tuple(tested_chunks), expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_non_intersected_chunks(self):
|
||
|
|
chunk_size = 4
|
||
|
|
|
||
|
|
cases = [
|
||
|
|
(tuple(range(12)), [[0,1,2,3], [4,5,6,7], [8,9,10,11]]),
|
||
|
|
(tuple(range(9)), [[0,1,2,3], [4,5,6,7], [8]]),
|
||
|
|
(tuple(range(10)), [[0,1,2,3], [4,5,6,7], [8,9]]),
|
||
|
|
(tuple(range(11)), [[0,1,2,3], [4,5,6,7], [8,9,10]]),
|
||
|
|
([], []),
|
||
|
|
(tuple(range(1)), [[0]]),
|
||
|
|
(tuple(range(4)), [[0,1,2,3]])
|
||
|
|
]
|
||
|
|
|
||
|
|
for tested, expected in cases:
|
||
|
|
tested_chunks = list(get_chunks(tested, chunk_size))
|
||
|
|
self.assertSequenceEqual(tested_chunks, expected)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|