tools/findbits_test.py

   1 #!/usr/bin/env python3
   2
   3 import unittest, sys, findbits
   4
   5 class TestFindBits(unittest.TestCase):
   6     def setUp(self):
   7         self.old_stdout = sys.stdout
   8         sys.stdout = OutputBuffer()
   9
  10     def tearDown(self):
  11         sys.stdout = self.old_stdout
  12
  13     INVERT_CASES = [
  14             ('10', '01'),
  15             ('', ''),
  16             ]
  17     def test_invert(self):
  18         self.commutative_test(findbits.invert, self.INVERT_CASES)
  19
  20     SEARCH_CASES = [
  21             ('1111', '10111101', ['Match at bit 2', '0<1111>0']),
  22             ('00', '10111101', ['Not found']),
  23             ]
  24     def test_search(self):
  25         for target, data, expected_fragments in self.SEARCH_CASES:
  26             sys.stdout.clear_buffer()
  27             findbits.search(target, data)
  28             for fragment in expected_fragments:
  29                 self.assertIn(fragment, sys.stdout.content)
  30
  31     BINSTRING_CASES = [
  32             (42, '101010'),
  33             (1, '1'),
  34             (0, ''),
  35             ]
  36     def test_binstring(self):
  37         self.unary_operation_test(findbits.binstring, self.BINSTRING_CASES)
  38
  39     REVERSE_CASES = [
  40             ('abc', 'cba'),
  41             ('', ''),
  42             ]
  43     def test_stringreverse(self):
  44         self.commutative_test(findbits.stringreverse, self.REVERSE_CASES)
  45
  46     def commutative_test(self, operation, cases):
  47         self.unary_operation_test(operation, cases)
  48         self.unary_operation_test(operation, map(reversed, cases))
  49
  50     def unary_operation_test(self, operation, cases):
  51         for case_in, case_out in cases:
  52             self.assertEqual(operation(case_in), case_out)
  53
  54
  55 class OutputBuffer(object):
  56     def __init__(self):
  57         self.clear_buffer()
  58
  59     def clear_buffer(self):
  60         self.content = ''
  61
  62     def write(self, data):
  63         self.content += data
  64
  65
  66 if __name__ == '__main__':
  67     unittest.main()