Mercurial > hg > Lemuriformes
diff tests/test_chunk.py @ 18:56596902e9ae default tip
add some setup + tests
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Sun, 10 Dec 2017 17:57:03 -0800 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/test_chunk.py Sun Dec 10 17:57:03 2017 -0800 @@ -0,0 +1,30 @@ +#!/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()