Back to home page

Project CMSSW displayed by LXR

 
 

    


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()