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