Lemuriformes: tests/test_uniques.py comparison

add some setup + tests

comparison

equal deleted inserted replaced

-:4793f99b73e0
+:56596902e9ae
+#!/usr/bin/env python
+"""
+test CSV file uniques counter
+"""
+import csv
+import os
+import shutil
+import tempfile
+import unittest
+from lemuriformes.uniques import uniques
+from StringIO import StringIO
+class TestUniques(unittest.TestCase):
+header = ['a', 'b', 'c']
+dataset = [[1,2,3],
+[1,1,2],
+[1,2,4]]
+def write_dataset(self, fp):
+writer = csv.writer(fp)
+writer.writerow(self.header)
+writer.writerows(self.dataset)
+fp.flush()
+def validate(self, uniq):
+"""validate `uniq` for object dataset"""
+# count them
+counts = {key: len(value)
+for key, value in uniq.items()}
+assert counts['a'] == 1
+assert counts['b'] == 2
+assert counts['c'] == 3
+def test_file(self):
+"""test reading uniques from a file"""
+tmpdir = tempfile.mkdtemp()
+try:
+# write test data
+dst = os.path.join(tmpdir, 'test.csv')
+with open(dst, 'w') as fp:
+self.write_dataset(fp)
+# determine uniques
+self.validate(uniques(dst))
+finally:
+shutil.rmtree(tmpdir, ignore_errors=True)
+def test_buffer(self):
+"""test using a StringIO buffer object"""
+# write test data to buffer
+buffer = StringIO()
+self.write_dataset(buffer)
+buffer.seek(0)
+# determine uniques
+self.validate(uniques(buffer))
+if __name__ == '__main__':
+unittest.main()

Mercurial > hg > Lemuriformes