예제 #1
0
class TestArray(unittest.TestCase):
    def setUp(self):
        self.c = Array(4, UBInt8("foo"))

    def test_trivial(self):
        pass

    def test_parse(self):
        self.assertEqual(self.c.parse(b"\x01\x02\x03\x04"), [1, 2, 3, 4])
        self.assertEqual(self.c.parse(b"\x01\x02\x03\x04\x05\x06"),
                         [1, 2, 3, 4])

    def test_build(self):
        self.assertEqual(self.c.build([5, 6, 7, 8]), b"\x05\x06\x07\x08")

    def test_build_oversized(self):
        self.assertRaises(ArrayError, self.c.build, [5, 6, 7, 8, 9])

    def test_build_undersized(self):
        self.assertRaises(ArrayError, self.c.build, [5, 6, 7])
예제 #2
0
class TestArray(unittest.TestCase):

    def setUp(self):
        self.c = Array(4, UBInt8("foo"))

    def test_trivial(self):
        pass

    def test_parse(self):
        self.assertEqual(self.c.parse(six.b("\x01\x02\x03\x04")), [1, 2, 3, 4])
        self.assertEqual(self.c.parse(six.b("\x01\x02\x03\x04\x05\x06")),
            [1, 2, 3, 4])

    def test_build(self):
        self.assertEqual(self.c.build([5, 6, 7, 8]), six.b("\x05\x06\x07\x08"))

    def test_build_oversized(self):
        self.assertRaises(ArrayError, self.c.build, [5, 6, 7, 8, 9])

    def test_build_undersized(self):
        self.assertRaises(ArrayError, self.c.build, [5, 6, 7])