def test_toImages(self): from lambdaimage.rdds.images import Images rdd = self.sc.parallelize([((0, 0), array([1])), ((0, 1), array([2])), ((1, 0), array([3])), ((1, 1), array([4]))]) data = Series(rdd) imgs = data.toImages() assert(isinstance(imgs, Images)) im = imgs.values().first() assert(allclose(im, [[1, 2], [3, 4]]))