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