def test_dtype(self):
foo = Foo()
foo.maybe_float_array = [1, 2, 3]
array_value = foo.maybe_float_array
self.assertIsInstance(array_value, numpy.ndarray)
self.assertEqual(array_value.dtype, numpy.dtype(float))
def test_setting_array_from_list(self):
foo = Foo()
test_list = [5, 6, 7, 8, 9]
foo.maybe_array = test_list
output_array = foo.maybe_array
self.assertIsInstance(output_array, numpy.ndarray)
self.assertEqual(output_array.dtype, numpy.dtype(int))
self.assertEqual(output_array.shape, (5, ))
self.assertTrue((output_array == test_list).all())
def test_image_to_array_rgb(self):
wximage = wx.Image(32, 64)
wximage.SetRGB(wx.Rect(0, 0, 32, 64), 0x44, 0x88, 0xcc)
array = image_to_array(wximage)
self.assertEqual(array.shape, (32, 64, 4))
self.assertEqual(array.dtype, np.dtype('uint8'))
self.assertTrue(np.all(array[:, :, 3] == 0xff))
self.assertTrue(np.all(array[:, :, 0] == 0x44))
self.assertTrue(np.all(array[:, :, 1] == 0x88))
self.assertTrue(np.all(array[:, :, 2] == 0xcc))
def test_image_to_array_rgba(self):
qimage = QImage(32, 64, QImage.Format.Format_ARGB32)
qimage.fill(QColor(0x44, 0x88, 0xcc, 0xee))
array = image_to_array(qimage)
self.assertEqual(array.shape, (64, 32, 4))
self.assertEqual(array.dtype, np.dtype('uint8'))
self.assertTrue(np.all(array[:, :, 0] == 0x44))
self.assertTrue(np.all(array[:, :, 1] == 0x88))
self.assertTrue(np.all(array[:, :, 2] == 0xcc))
self.assertTrue(np.all(array[:, :, 3] == 0xee))
def test_set_structured_dtype(self):
color = Color(rgba=(0.4, 0.2, 0.6, 1.0))
arr = np.array(
[(0.4, 0.2, 0.6, 1.0)],
dtype=np.dtype([
('red', float),
('green', float),
('blue', float),
('alpha', float),
]),
)
color_class = ColorClass(color=arr[0])
self.assertEqual(color_class.color, color)
def test_image_to_array_rgba(self):
wximage = wx.Image(32, 64)
wximage.SetRGB(wx.Rect(0, 0, 32, 64), 0x44, 0x88, 0xcc)
wximage.InitAlpha()
wximage.SetAlpha(np.full((32 * 64, ), 0xee, dtype='uint8'))
array = image_to_array(wximage)
self.assertEqual(array.shape, (32, 64, 4))
self.assertEqual(array.dtype, np.dtype('uint8'))
self.assertTrue(np.all(array[:, :, 0] == 0x44))
self.assertTrue(np.all(array[:, :, 1] == 0x88))
self.assertTrue(np.all(array[:, :, 2] == 0xcc))
self.assertTrue(np.all(array[:, :, 3] == 0xee))
def test_init_array_structured_dtype(self):
""" Test if "typical" RGBA structured array value works. """
arr = np.array(
[(0.4, 0.2, 0.6, 1.0)],
dtype=np.dtype([
('red', float),
('green', float),
('blue', float),
('alpha', float),
]),
)
trait = PyfaceColor(arr[0])
self.assertEqual(trait.default_value, (Color, (), {
'rgba': (0.4, 0.2, 0.6, 1.0)
}))
