File indexing completed on 2024-11-25 02:29:50
0001 import unittest
0002 import os
0003 import shutil
0004
0005 from .counter import Counter
0006
0007 class CounterTestCase(unittest.TestCase):
0008
0009 def test_simple(self):
0010 c = Counter('Test')
0011 c.register('a')
0012 c.register('b')
0013 c.inc('a')
0014 self.assertEqual(c['a'], ['a', 1])
0015 self.assertEqual(c['b'], ['b', 0])
0016 c.inc('a')
0017 self.assertEqual(c['a'], ['a', 2])
0018
0019 def test_add(self):
0020 c = Counter('Test')
0021 c.register('a')
0022 c.register('b')
0023 c.inc('a')
0024 d = Counter('Test')
0025 d.register('a')
0026 d.register('b')
0027 d.inc('a')
0028 d.inc('b')
0029 d += c
0030 self.assertEqual(d['a'], ['a', 2])
0031 self.assertEqual(d['b'], ['b', 1])
0032
0033 def test_bad_add(self):
0034 c = Counter('Test')
0035 c.register('a')
0036 c.register('b')
0037 c.inc('a')
0038 d = Counter('Test')
0039 d.register('b')
0040 self.assertRaises(ValueError, d.__iadd__, c)
0041
0042 def test_write(self):
0043 c = Counter('Test')
0044 c.register('a')
0045 c.register('b')
0046 c.inc('a')
0047 dirname = 'test_dir'
0048 if os.path.exists(dirname):
0049 shutil.rmtree(dirname)
0050 os.mkdir(dirname)
0051 c.write(dirname)
0052 shutil.rmtree(dirname)
0053
0054
0055
0056
0057
0058
0059 if __name__ == '__main__':
0060 unittest.main()