old_contents: #!/usr/bin/env python
  
  """Unittest for kegboard module"""
  
  import os
  import unittest
  import struct
  
  from pykeg.hw.kegboard import kegboard
  
  TESTDATA_PATH = os.path.join(os.path.dirname(kegboard.__file__), 'testdata')
  CAP_FILE = os.path.join(TESTDATA_PATH, 'one_flow_active.bin')
  
  
  class MessageTestCase(unittest.TestCase):
    def testMessageCreate(self):
      hello_bytes = kegboard.KBSP_PREFIX + '\x01\x00\x04\x00\x01\x02\x03\x00\x3f\x29\r\n'
      m = kegboard.HelloMessage(bytes=hello_bytes)
      self.assertEqual(m.protocol_version, 3)
  
      m = kegboard.GetMessageForBytes(hello_bytes)
      self.assertEqual(m.protocol_version, 3)
      print m
  
  
  class KegboardReaderTestCase(unittest.TestCase):
    def testBasicUse(self):
      fd = open(CAP_FILE, 'rb')
      kbr = kegboard.KegboardReader(fd)
  
      # read the first 8 messages
      messages = []
      for i in xrange(8):
        messages.append(kbr.GetNextMessage())
      print 'messages:'
      print '\n'.join('  %s' % msg for msg in messages)
  
      hello_message = kegboard.HelloMessage()
      hello_message.SetValue('protocol_version', 3)
      self.assertEqual(messages[0], hello_message)
  
      onewire_message = kegboard.OnewirePresenceMessage()
      onewire_message.SetValue('device_id', 0)
      self.assertEqual(messages[1], onewire_message)
  
      message_bytes = messages[2].ToBytes()
      new_message = kegboard.GetMessageForBytes(message_bytes)
      self.assertEqual(message_bytes, new_message.ToBytes())
  
    def testAgainstBogusData(self):
      pass
  
  if __name__ == '__main__':
    unittest.main()
  
new_contents: #!/usr/bin/env python
  
  """Unittest for kegboard module"""
  
  import os
  import unittest
  import struct
  
  from pykeg.hw.kegboard import kegboard
  
  TESTDATA_PATH = os.path.join(os.path.dirname(kegboard.__file__), 'testdata')
  CAP_FILE = os.path.join(TESTDATA_PATH, 'one_flow_active.bin')
  
  
  class MessageTestCase(unittest.TestCase):
    def testMessageCreate(self):
      hello_bytes = kegboard.KBSP_PREFIX + '\x01\x00\x04\x00\x01\x02\x03\x00\x3f\x29\r\n'
      m = kegboard.HelloMessage(bytes=hello_bytes)
      self.assertEqual(m.protocol_version, 3)
  
      m = kegboard.GetMessageForBytes(hello_bytes)
      self.assertEqual(m.protocol_version, 3)
      print m
  
  
  class KegboardReaderTestCase(unittest.TestCase):
    def testBasicUse(self):
      fd = open(CAP_FILE, 'rb')
      kbr = kegboard.KegboardReader(fd)
  
      # read the first 8 messages
      messages = []
      for i in xrange(8):
        messages.append(kbr.GetNextMessage())
      print 'messages:'
      print '\n'.join('  %s' % msg for msg in messages)
  
      hello_message = kegboard.HelloMessage()
      hello_message.SetValue('protocol_version', 3)
      self.assertEqual(messages[0], hello_message)
  
      onewire_message = kegboard.OnewirePresenceMessage()
      onewire_message.SetValue('device_id', 0)
      self.assertEqual(messages[1], onewire_message)
  
      message_bytes = messages[2].ToBytes()
      new_message = kegboard.GetMessageForBytes(message_bytes)
      self.assertEqual(message_bytes, new_message.ToBytes())
  
    def testAgainstBogusData(self):
      pass
  
  if __name__ == '__main__':
    import logging
    logging.basicConfig()
    unittest.main()
  
subject: Fix logging during unittest.