Completed
Pull Request — master (#21)
by Philip
01:10
created

TestChunks.test_chunks_sequence()   A

Complexity

Conditions 2

Size

Total Lines 9

Duplication

Lines 0
Ratio 0 %

Importance

Changes 1
Bugs 1 Features 0
Metric Value
cc 2
c 1
b 1
f 0
dl 0
loc 9
rs 9.6666
1
import unittest
2
3
from foil.iteration import chunks
4
5
6
class TestChunks(unittest.TestCase):
7
    def test_chunks_generator(self):
8
        gen = range(0, 8)
9
        chunksize = 3
10
        grouped = chunks(gen, chunksize=chunksize)
11
12
        expected = [(0, 1, 2), (3, 4, 5), (6, 7)]
13
        result = list(tuple(g) for g in grouped)
14
15
        self.assertEqual(expected, result)
16
17
    def test_chunks_sequence(self):
18
        seq = list(range(0, 9))
19
        chunksize = 3
20
        grouped = chunks(seq, chunksize=chunksize)
21
22
        expected = [(0, 1, 2), (3, 4, 5), (6, 7, 8)]
23
        result = [tuple(g) for g in grouped]
24
25
        self.assertEqual(expected, result)
26