mirror of
https://github.com/wxWidgets/Phoenix.git
synced 2025-09-05 17:30:26 +02:00
c2e5a59a8f
* Use wxPyBuffer with wx.Image * Return bytearrays from GetData and GetAlpha * Return memoryviews from GetDataBuffer and GetAlphaBuffer * Add some notes about these changes to the migration guide git-svn-id: https://svn.wxwidgets.org/svn/wx/wxPython/Phoenix/trunk@71291 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
254 lines
7.1 KiB
Python
254 lines
7.1 KiB
Python
import imp_unittest, unittest
|
|
import wtc
|
|
import wx
|
|
import os
|
|
|
|
pngFile = os.path.join(os.path.dirname(__file__), 'toucan.png')
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def makeBuf(w, h, bpp=1, init=0):
|
|
"Make a simple buffer for testing with"
|
|
|
|
# Apparently array objects do not implement the new buffer protocol...
|
|
#import array
|
|
#buf = array.array('B', [init] * (w*h*bpp))
|
|
|
|
buf = bytearray([init] * (w*h*bpp))
|
|
return buf
|
|
|
|
|
|
class image_Tests(wtc.WidgetTestCase):
|
|
|
|
def test_imageCtor1(self):
|
|
img = wx.Image()
|
|
self.assertTrue(not img.IsOk())
|
|
img.Create(100,100)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor2(self):
|
|
img = wx.Image(100,100)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor3(self):
|
|
img = wx.Image(wx.Size(100,100))
|
|
self.assertTrue(img.IsOk())
|
|
img = wx.Image((100,100))
|
|
self.assertTrue(img.IsOk())
|
|
|
|
|
|
def test_imageCtor4(self):
|
|
w = h = 10
|
|
buf = makeBuf(w,h,3)
|
|
img = wx.Image(w, h, buf)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor5(self):
|
|
w = h = 10
|
|
buf = makeBuf(w,h,3)
|
|
alpha = makeBuf(w,h)
|
|
img = wx.Image(w, h, buf, alpha)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor4b(self):
|
|
w = h = 10
|
|
buf = makeBuf(w,h,3)
|
|
img = wx.Image((w, h), buf)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor4c(self):
|
|
w = h = 10
|
|
buf = makeBuf(w,h,3)
|
|
with self.assertRaises(ValueError):
|
|
# should be an exception here because the buffer is the wrong size
|
|
img = wx.Image((w, h+1), buf)
|
|
|
|
def test_imageCtor5b(self):
|
|
w = h = 10
|
|
buf = makeBuf(w,h,3)
|
|
alpha = makeBuf(w,h)
|
|
img = wx.Image((w, h), buf, alpha)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
|
|
def test_imageCtor6(self):
|
|
img = wx.Image(pngFile, wx.BITMAP_TYPE_PNG)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor7(self):
|
|
img = wx.Image(pngFile, 'image/png')
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor8(self):
|
|
data = open(pngFile, 'rb').read()
|
|
import StringIO
|
|
stream = StringIO.StringIO(data)
|
|
img = wx.Image(stream, wx.BITMAP_TYPE_PNG)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
def test_imageCtor9(self):
|
|
data = open(pngFile, 'rb').read()
|
|
import StringIO
|
|
stream = StringIO.StringIO(data)
|
|
img = wx.Image(stream, 'image/png')
|
|
self.assertTrue(img.IsOk())
|
|
|
|
|
|
def test_imageSetData1(self):
|
|
w = h = 10
|
|
img = wx.Image(w,h)
|
|
buf = makeBuf(w,h,3, init=2)
|
|
img.SetData(buf)
|
|
self.assertTrue(img.IsOk())
|
|
self.assertTrue(img.GetRed(1,1) == 2)
|
|
|
|
def test_imageSetData2(self):
|
|
w = h = 10
|
|
img = wx.Image(1,1)
|
|
buf = makeBuf(w,h,3, init=2)
|
|
img.SetData(buf, w, h)
|
|
self.assertTrue(img.IsOk())
|
|
self.assertTrue(img.GetRed(1,1) == 2)
|
|
|
|
def test_imageSetAlpha1(self):
|
|
w = h = 10
|
|
img = wx.Image(w,h)
|
|
buf = makeBuf(w,h, init=2)
|
|
img.SetAlpha(buf)
|
|
self.assertTrue(img.IsOk())
|
|
self.assertTrue(img.GetRed(1,1) == 0)
|
|
self.assertTrue(img.GetAlpha(1,1) == 2)
|
|
|
|
def test_imageGetData(self):
|
|
img = wx.Image(pngFile)
|
|
data = img.GetData()
|
|
self.assertEqual(len(data), img.Width * img.Height * 3)
|
|
self.assertTrue(isinstance(data, bytearray))
|
|
|
|
def test_imageGetAlpha(self):
|
|
img = wx.Image(pngFile)
|
|
data = img.GetAlpha()
|
|
self.assertEqual(len(data), img.Width * img.Height)
|
|
self.assertTrue(isinstance(data, bytearray))
|
|
|
|
|
|
def test_imageGetDataBuffer(self):
|
|
w = h = 10
|
|
img = wx.Image(w, h)
|
|
self.assertTrue(img.IsOk())
|
|
data = img.GetDataBuffer()
|
|
self.assertTrue(isinstance(data, memoryview))
|
|
data[0] = '\1'
|
|
data[1] = '\2'
|
|
data[2] = '\3'
|
|
self.assertEqual(1, img.GetRed(0,0))
|
|
self.assertEqual(2, img.GetGreen(0,0))
|
|
self.assertEqual(3, img.GetBlue(0,0))
|
|
|
|
def test_imageGetAlphaDataBuffer(self):
|
|
w = h = 10
|
|
img = wx.Image(w, h)
|
|
img.InitAlpha()
|
|
self.assertTrue(img.IsOk())
|
|
data = img.GetAlphaBuffer()
|
|
self.assertTrue(isinstance(data, memoryview))
|
|
data[0] = '\1'
|
|
data[1] = '\2'
|
|
data[2] = '\3'
|
|
self.assertEqual(1, img.GetAlpha(0,0))
|
|
self.assertEqual(2, img.GetAlpha(1,0))
|
|
self.assertEqual(3, img.GetAlpha(2,0))
|
|
|
|
|
|
def test_imageSetDataBuffer1(self):
|
|
w = h = 10
|
|
img = wx.Image(w,h)
|
|
buf = makeBuf(w,h,3)
|
|
img.SetDataBuffer(buf)
|
|
buf[0] = 1
|
|
buf[1] = 2
|
|
buf[2] = 3
|
|
self.assertEqual(1, img.GetRed(0,0))
|
|
self.assertEqual(2, img.GetGreen(0,0))
|
|
self.assertEqual(3, img.GetBlue(0,0))
|
|
|
|
def test_imageSetDataBuffer2(self):
|
|
w = h = 10
|
|
img = wx.Image(1,1)
|
|
buf = makeBuf(w,h,3)
|
|
img.SetDataBuffer(buf, w, h)
|
|
buf[0] = 1
|
|
buf[1] = 2
|
|
buf[2] = 3
|
|
self.assertEqual(1, img.GetRed(0,0))
|
|
self.assertEqual(2, img.GetGreen(0,0))
|
|
self.assertEqual(3, img.GetBlue(0,0))
|
|
|
|
def test_imageSetAlphaBuffer(self):
|
|
w = h = 10
|
|
img = wx.Image(w,h)
|
|
buf = makeBuf(w,h)
|
|
img.SetAlphaBuffer(buf)
|
|
buf[0] = 1
|
|
buf[1] = 2
|
|
buf[2] = 3
|
|
self.assertEqual(1, img.GetAlpha(0,0))
|
|
self.assertEqual(2, img.GetAlpha(1,0))
|
|
self.assertEqual(3, img.GetAlpha(2,0))
|
|
|
|
|
|
def test_imageNestedClasses(self):
|
|
rgb = wx.Image.RGBValue(1,2,3)
|
|
self.assertEqual(rgb.red, 1)
|
|
self.assertEqual(rgb.green, 2)
|
|
self.assertEqual(rgb.blue, 3)
|
|
rgb.red = 4
|
|
rgb.green = 5
|
|
rgb.blue = 6
|
|
|
|
hsv = wx.Image.HSVValue(1.1, 1.2, 1.3)
|
|
self.assertEqual(hsv.hue, 1.1)
|
|
self.assertEqual(hsv.saturation, 1.2)
|
|
self.assertEqual(hsv.value, 1.3)
|
|
hsv.hue = 2.1
|
|
hsv.saturation = 2.2
|
|
hsv.value = 2.3
|
|
|
|
|
|
def test_imageRGBHSV(self):
|
|
rgb = wx.Image.RGBValue(1,2,3)
|
|
hsv = wx.Image.RGBtoHSV(rgb)
|
|
rgb = wx.Image.HSVtoRGB(hsv)
|
|
self.assertEqual(rgb.red, 1)
|
|
self.assertEqual(rgb.green, 2)
|
|
self.assertEqual(rgb.blue, 3)
|
|
|
|
|
|
def test_imageProperties(self):
|
|
img = wx.Image(pngFile)
|
|
self.assertTrue(img.IsOk())
|
|
img.Width
|
|
img.Height
|
|
img.MaskRed
|
|
img.MaskGreen
|
|
img.MaskBlue
|
|
img.Type
|
|
|
|
|
|
def test_imageMethodChain(self):
|
|
img = wx.Image(100,100).Rescale(75,75).Resize((100,100), (0,0), 40,60,80)
|
|
self.assertTrue(img.IsOk())
|
|
|
|
|
|
def test_imageOtherStuff(self):
|
|
img = wx.Image(pngFile)
|
|
self.assertTrue(img.IsOk())
|
|
r, g, b = img.FindFirstUnusedColour()
|
|
r, g, b = img.GetOrFindMaskColour()
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|