code.kerkeslager.com Git - sandbox/blob - serial/test_binary.py

   1 import unittest
   2
   3 from serial import binary
   4
   5 EXAMPLE_REPRESENTATIONS = [
   6     (binary.TaggedObject(binary.TAG_NULL, None), b'\x00'),
   7     (binary.TaggedObject(binary.TAG_TRUE, True), b'\x01'),
   8     (binary.TaggedObject(binary.TAG_FALSE, False), b'\x02'),
   9     (binary.TaggedObject(binary.TAG_UINT8, 7), b'\x03\x07'),
  10     (binary.TaggedObject(binary.TAG_UINT16, 7), b'\x04\x00\x07'),
  11     (binary.TaggedObject(binary.TAG_UINT32, 7), b'\x05\x00\x00\x00\x07'),
  12     (binary.TaggedObject(binary.TAG_UINT64, 7), b'\x06\x00\x00\x00\x00\x00\x00\x00\x07'),
  13     (binary.TaggedObject(binary.TAG_INT8, 7), b'\x10\x07'),
  14     (binary.TaggedObject(binary.TAG_INT16, 7), b'\x11\x00\x07'),
  15     (binary.TaggedObject(binary.TAG_INT32, 7), b'\x12\x00\x00\x00\x07'),
  16     (binary.TaggedObject(binary.TAG_INT64, 7), b'\x13\x00\x00\x00\x00\x00\x00\x00\x07'),
  17     (binary.TaggedObject(binary.TAG_UINT8, 254), b'\x03\xfe'),
  18     (binary.TaggedObject(binary.TAG_UINT16, 65534), b'\x04\xff\xfe'),
  19     (binary.TaggedObject(binary.TAG_UINT32, 4294967294), b'\x05\xff\xff\xff\xfe'),
  20     (binary.TaggedObject(binary.TAG_UINT64, 18446744073709551614), b'\x06\xff\xff\xff\xff\xff\xff\xff\xfe'),
  21     (binary.TaggedObject(binary.TAG_INT8, -2), b'\x10\xfe'),
  22     (binary.TaggedObject(binary.TAG_INT16, -2), b'\x11\xff\xfe'),
  23     (binary.TaggedObject(binary.TAG_INT32, -2), b'\x12\xff\xff\xff\xfe'),
  24     (binary.TaggedObject(binary.TAG_INT64, -2), b'\x13\xff\xff\xff\xff\xff\xff\xff\xfe'),
  25     (binary.TaggedObject(binary.TAG_BINARY, b'\xde\xad\xbe\xef'), b'\x20\x00\x00\x00\x04\xde\xad\xbe\xef'),
  26     (binary.TaggedObject(binary.TAG_UTF8, 'Lol!'), b'\x21\x00\x00\x00\x04Lol!'),
  27     (binary.TaggedObject(binary.TAG_UTF16, 'かわ'), b'\x22\x00\x00\x00\x06\xff\xfeK0\x8f0'),
  28     (binary.TaggedObject(binary.TAG_UTF32, '漢'), b'\x23\x00\x00\x00\x08\xff\xfe\x00\x00"o\x00\x00'),
  29     (
  30         binary.TaggedObject(
  31             binary.TAG_TUPLE,
  32             (
  33                 binary.TaggedObject(
  34                     binary.TAG_TRUE,
  35                     True,
  36                 ),
  37                 binary.TaggedObject(
  38                     binary.TAG_UINT8,
  39                     7,
  40                 ),
  41             ),
  42         ),
  43         b'\x30\x00\x00\x00\x03\x01\x03\x07'
  44     ),
  45 ]
  46
  47 class SerializeTests(unittest.TestCase):
  48     def test_serialize(self):
  49         for tagged_object, expected in EXAMPLE_REPRESENTATIONS:
  50             actual = binary.serialize(tagged_object)
  51             self.assertEqual(expected, actual)
  52
  53 class DeserializeTests(unittest.TestCase):
  54     def test_deserialize(self):
  55         for expected, representation in EXAMPLE_REPRESENTATIONS:
  56             actual = binary.deserialize(representation)
  57             self.assertEqual(expected, actual)
  58
  59 unittest.main()