Lemuriformes: tests/test_chunk.py comparison

add some setup + tests

comparison

equal deleted inserted replaced

-:4793f99b73e0
+:56596902e9ae
+#!/usr/bin/env python
+"""
+test data chunking
+"""
+import os
+import unittest
+from lemuriformes.chunk import chunk
+class TestChunking(unittest.TestCase):
+def test_range(self):
+"""test basic chunking with range function"""
+data = list(range(64))
+chunked = list(chunk(data, 10))
+assert len(chunked) == 7
+sizes = set([len(c) for c in
+chunked[:-1]])
+assert len(sizes) == 1
+assert sizes.pop() == 10
+reconstructed = []
+for _chunk in chunked:
+reconstructed.extend(_chunk)
+assert reconstructed == list(range(64))
+if __name__ == '__main__':
+unittest.main()

Mercurial > hg > Lemuriformes