class TestDrawing(TestCase): def setUp(self): self.drawing = Drawing(DrawingType.RECT, 0, [0, 0, 1, 1]) def test_encoding_decoding_are_equal(self): bytes_array = self.drawing.encode() other = Drawing.decode_drawing(bytes_array) self.assertEqual(self.drawing, other)
class TestDrawing(TestCase): def setUp(self): self.drawing = Drawing(DrawingType.RECT, 0, "#000000", [0, 0, 1, 1]) self.text_drawing = Drawing(DrawingType.TEXT, 0, "#000000", [0, 0, 0, 0], "testing") def test_encoding_decoding_are_equal(self): bytes_array = self.drawing.encode() decoded, decoded_length = Drawing.decode_drawing(bytes_array) self.assertEqual(len(bytes_array), decoded_length) self.assertEqual(self.drawing, decoded) def test_decoding_multiple_drawings(self): bytes_array = self.drawing.encode() + self.text_drawing.encode() decoded = Drawing.decode_drawings(bytes_array) self.assertEqual(decoded[0], self.drawing) self.assertEqual(decoded[1], self.text_drawing) def test_decoding_text(self): bytes_array = self.text_drawing.encode() decoded, decoded_length = Drawing.decode_drawing(bytes_array) self.assertEqual(len(bytes_array), decoded_length) self.assertEqual(self.text_drawing, decoded)