def test_parcel_write(self): parcelmap = np.zeros(self.img3.shape) parcelmap[0,0,0] = 1 parcelmap[1,1,1] = 1 parcelmap[2,2,2] = 1 parcelmap[1,2,1] = 2 parcelmap[2,3,2] = 2 parcelmap[0,1,0] = 2 parcelseq = (0, 1, 2, 3) expected = [np.product(self.img3.shape) - 6, 3, 3, 0] iterator = gen.parcels(parcelmap, labels=parcelseq) for i, s in enumerate(iterator): value = np.arange(expected[i]) self.img3[s] = value iterator = gen.parcels(parcelmap, labels=parcelseq) for i, pair in enumerate(gen.data_generator(self.img3, iterator)): s, d = pair self.assertEqual((expected[i],), d.shape) assert_equal(d, np.arange(expected[i])) iterator = gen.parcels(parcelmap) for i, s in enumerate(iterator): value = np.arange(expected[i]) self.img3[s] = value iterator = gen.parcels(parcelmap) for i, pair in enumerate(gen.data_generator(self.img3, iterator)): s, d = pair self.assertEqual((expected[i],), d.shape) assert_equal(d, np.arange(expected[i]))
def test_parcel(self): parcelmap = np.zeros(self.img3.shape) parcelmap[0,0,0] = 1 parcelmap[1,1,1] = 1 parcelmap[2,2,2] = 1 parcelmap[1,2,1] = 2 parcelmap[2,3,2] = 2 parcelmap[0,1,0] = 2 parcelseq = (0, 1, 2, 3) expected = [np.product(self.img3.shape) - 6, 3, 3, 0] iterator = gen.data_generator(self.img3, gen.parcels(parcelmap, labels=parcelseq)) for i, pair in enumerate(iterator): s, d = pair self.assertEqual((expected[i],), d.shape) iterator = gen.data_generator(self.img3, gen.parcels(parcelmap)) for i, pair in enumerate(iterator): s, d = pair self.assertEqual((expected[i],), d.shape)
def test_parcel_copy(self): parcelmap = np.zeros(self.img3.shape) parcelmap[0,0,0] = 1 parcelmap[1,1,1] = 1 parcelmap[2,2,2] = 1 parcelmap[1,2,1] = 2 parcelmap[2,3,2] = 2 parcelmap[0,1,0] = 2 parcelseq = (0, 1, 2, 3) expected = [np.product(self.img3.shape) - 6, 3, 3, 0] iterator = gen.parcels(parcelmap, labels=parcelseq) tmp = Image(np.asarray(self.img3), self.img3.coordmap) gen_parcels = gen.parcels(parcelmap, labels=parcelseq) new_iterator = gen.data_generator(tmp, gen_parcels) for i, slice_ in enumerate(new_iterator): self.assertEqual((expected[i],), slice_[1].shape)